[Algorithm] 기초 문제 / 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

 

▶ 출력

#1 7

#2 6

 

---

 

▷ 내가 짜본 코드

import sys
sys.stdin = open('input.txt', 'rt')

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(f'#{t+1} {a[k-1]}')

 

너무 기본적인 구현이라 로직 자체는 헷갈릴 게 없었는데 문제는 의외로 입력을 받아오는 데 있었다.

 

케이스가 두 개라서 그 두개의 값을 한 줄씩 읽도록 반복문을 돌리는 부분이 어려워서 한참을 헤맸다. 

이 부분은 계속 반복해서 익숙해 지는 수밖에 없을 것 같다.