[구현력 기르기] - k번째 수

 k번째 수

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

 

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

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

 

▣ 입력예제 1

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

#1 7
#2 6

 

입력예제1 해설 :
case 1 : 2 7 3 8의 숫자 중 오름차순 정렬 했을 때 3번째 숫자는 7이다.
case 2 : 8 16 6 6 17 3 10 11의 숫자 중 오름차순 정렬 했을 때 3번째 숫자는 6이다.

 

 

 

 

 

나의 코드

T=int(input())


for i in range(T):
    n,s,e,k=map(int,input().split())
    arr=list(map(int,input().split()))
    temp=arr[s-1:e]
    temp.sort()
    print('#',i+1 , temp[k-1])

입력받은 T개의 테스트케이스 갯수만큼 입력 받은 그 횟수만큼 반복을 한다.

해당 n개의 숫자를 arr리스트에 저장한 뒤 슬라이싱 처리를 해서 필요한 리스트를 temp에 다시 저장한다.

그후 sort함수로 정렬을 해서 K번쨰 순서를 출력

 

 

정답코드

import sys
sys.stdin=open("input.txt", "r")
T=int(input())
for t in range(T):
    n, s, e, k=map(int, input().split())
    a=list(map(int, input().split()))
    a=a[s-1:e]
    a.sort()
    print("#%d %d" %(t+1, a[k-1]))

 

나의 코드와 거의 일치 하지만 출력하는 코드가 다르다.

정답코드의 코드가 더 일반적으로 좋은 코드라 생각 된다..!
기억해야쥥