[프로그래머스-Lv1] K번째수 / Python

코딩테스트 연습 - K번째수

import sys

input = sys.stdin.readline
n = int(input())
array = list(map(int, input().split()))
commands = [list(map(int, input().split())) for _ in range(n)]


def solution(array, commands):
    answer = []
    for i in commands:
        temp = array[i[0] - 1 : i[1]]
        temp.sort()
        answer.append(temp[i[2] - 1])
    return answer


print(solution(array, commands))

 

해설

주어진 범위로 리스트를 잘라 답을 추출한다.