Python | K번째 수

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

코드 💻

#풀이1 - 슬라이싱

T = int(input())
# T = 2
for i in range(0, T):
    # N개의 숫자
    # s번째부터 e번째 까지의 수
    # k번째로 나타나는 숫자

    N, s, e, k = map(int, input().split()) # N, s, e, k = 6,2,5,3
    num = list(map(int, input().split())) # num = [5,2,7,3,8,9]

    answer = num[s-1:e]
    answer.sort()
    print(f'#{i+1} ',answer[k-1])

#풀이2 - for문

T = int(input())
# T = 2
for i in range(0, T):
    # N개의 숫자
    # s번째부터 e번째 까지의 수
    # k번째로 나타나는 숫자

    N, s, e, k = map(int, input().split()) # N, s, e, k = 6,2,5,3
    num = list(map(int, input().split())) # num = [5,2,7,3,8,9]

    answer = []
    for i in range(s-1, e):
    answer.append(num[i])
    answer.sort()
    print(f'#{i+1} ',answer[k-1])

출처 : 한국정보올림피아드