(Python) K번째수 (프로그래머스 Lv.1)/한줄코딩

K번째수 (프로그래머스 Lv.1)

https://school.programmers.co.kr/learn/courses/30/lessons/42748

프로그래머스

 

내 코드 

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

주의 할점은 0번째가 아니라 1번째부터 세기 시작한다는 점이다.

간단하게 썼다고 생각했지만 베스트코드를 보니 이런게 나왔다..나도 lambda고수가 되고싶다.

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