[Python/알고리즘] K번째 수

문제

N개의 숫자로 이루어진 숫자열이 주어지면 해당 숫자열중에서 s번째부터 e번째 까지의 수를 오름 차순 정렬했을 때 k번째로 나타나는 숫자를 출력하는 프로그램을 작성하세요.

 

입력

첫 번째 줄에 테스트 케이스 T(1<=T<=10)이 주어집니다. 각 케이스별 첫 번째 줄은 자연수 N(5<=N<=500), s, e, k가 차례로 주어진다. 두 번째 줄에 N개의 숫자가 차례로 주어진다.

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

 

출력

각 케이스별 k번째 수를 아래 출력예제와 같이 출력하세요.

#1 7 #2 6

 

답

T = int(input())

for i in range(T):
    N, s, e, k = map(int, input().split())

    li = list(map(int, input().split()))

    li = li[s - 1:e]
    li.sort()

    print(f"#{i + 1} {li[k - 1]}")

💡 list(map(int, input().split()))

: 입력 받은 값을 int형으로 리스트에 저장.