[알고리즘] K번째 수
(문제) K번째 수
N개의 숫자로 이루어진 숫자열이 주어지면 해당 숫자열중에서 s번째부터 e번째 까지의 수를 오름 차순 정렬했을 때 k번째로 나타나는 숫자를 출력하는 프로그램을 작성하세요.
입력설명
첫 번째 줄에 테스트 케이스 T(1<=T<=10)이 주어집니다.
각 케이스별 첫 번째 줄은 자연수 N(5<=N<=500), s, e, k가 차례로 주어진다.
두 번째 줄에 N개의 숫자가 차례로 주어진다.
출력설명
각 케이스별 k번째 수를 아래 출력예제와 같이 출력하세요.
테스트케이스
입력예제 | 출력예제 |
---|---|
2 6 2 5 3 5 2 7 3 8 9 15 3 10 3 4 15 8 16 6 6 17 3 10 11 18 7 14 7 15 |
#1 7 #2 6 |
해결방법
- 숫자들을 받아서 리스트 슬라이싱을 통해 리스트를 자르고, 전부 잘랐으면 그것을 정렬 시켜 k 번째를 출력하면 된다.
코드
for i in range(int(input())):
n, s, e, k = map(int, input().split())
num = list(map(int, input().split()))
print(f"#{i+1} {sorted(num[s-1:e])[k-1]}")
댓글남기기