프로그래머스 / 코딩 테스트 / K번째 수

문제 설명

 

 

문제 해결

 

문제에서 요구하는 데로 commands를 이용해 array를 인덱싱하여 정렬후 원소를 추출하면 된다.

comprehension을 이용하면 한 줄로 풀 수 있다.

 

def solve(array, commands):
    return [sorted(array[command[0]-1:command[1]])[command[2]-1] for command in commands]   

def solution(array, commands):
    answer = solve(array, commands)
    return answer