문제 / K번째 수 / Python

 

문제 제목: 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

 

 

입력예제 해설 :

case 1 : 2 7 3 8의 숫자 중 오름차순 정렬 했을 때 3번째 숫자는 7이다.

case 2 : 8 16 6 6 17 3 10 11의 숫자 중 오름차순 정렬 했을 때 3번째 숫자는 6이다

 

---

나의 풀이

import sys
T = int(input())

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

    new_arr = sorted(arr[s-1:e])
    print("#%d %d"%(i+1, new_arr[k-1]))

 

 

---

※ 알아야 할 것

- 프로그래머스 1단계 K번째 수와 똑같은 문제이다.