[Python] K번째 수

1. 문제

 

 

 

2. 풀이 과정

 

방법 1) sorted 함수를 써서 정렬된 결과를 반환받음

def solution(array, commands):
    answer = []
    for command in commands:
        i, j, k = command
        temp = sorted(array[i-1:j])
        answer.append(temp[k-1])
    return answer

 

방법 2) 방법1과 같은 원리이며 사용한 함수만 다르다. 아래는 map, lambda 함수를 이용하여 더 짧게 표현하였다.

def solution(array, commands):
    answer = [*(map(lambda x : sorted(array[x[0]-1:x[1]])[x[2]-1], commands))]
    return answer