[알고리즘] 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

해결방법

  1. 숫자들을 받아서 리스트 슬라이싱을 통해 리스트를 자르고, 전부 잘랐으면 그것을 정렬 시켜 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]}")

댓글남기기