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

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

프로그래머스

풀이

sorted와 리스트 인덱싱을 사용해 답을 구할 수 있습니다.

 

for문을 map함수로 대체하면 한줄로 답을 구할 수 있습니다.

def solution(array, commands):
    answer = []
    #for idx, command in enumerate(commands):
            #answer.append(sorted(array[command[0]-1:command[1]])[command[2]-1])
    
    return list(map(lambda x : sorted(array[x[0]-1:x[1]])[x[2]-1], commands))
    #return answer