하루한줄 코딩일기

[인프런 - 파이썬 알고리즘 문제풀이] 2.2 K번째 수 본문

Algorithm

[인프런 - 파이썬 알고리즘 문제풀이] 2.2 K번째 수

jjuha 2022. 1. 24. 15:50

인프런 > 파이썬 알고리즘 문제풀이

섹션 2.2 K번째 수

👊 내 문제 풀이

첫 줄은 케이스의 개수를 알려준다. 첫 줄을 읽어와서 numOfCase라는 변수에 저장한다.

케이스 개수 동안 n, s, e, k에 해당하는 숫자들을 저장하고, arr 리스트에 그 다음 줄인 숫자열을 저장한다.

숫자열 arr을 s부터 e번째 수까지 자른 후 정렬하고, 정답을 출력한다.

numOfCase=int(input())	#케이스의 개수

for i in range(numOfCase):
    n, s, e, k = map(int, input().split())	#띄어쓰기를 기준으로 읽어들임
    arr = list(map(int, input().split()))	#리스트에 숫자열 저장
    arr = arr[s-1:e]	#s번째부터 e번째까지의 수
    arr.sort()	#오름차순 정렬
    print("#%d %d" %(i+1, arr[k-1]))

채점 결과

Comments