프로그래머스(programmers) K번째수 python 정답 [정렬]

프로그래머스(programmers) K번째수 python 정답 [정렬]

문제 링크

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

프로그래머스

문제 정답

입출력 예

array	commands	return
[1, 5, 2, 6, 3, 7, 4]	[[2, 5, 3], [4, 4, 1], [1, 7, 3]]	[5, 6, 3]

 

정답 코드

def solution(array, commands):
    def mapCommand(command):
        [s,e,k] = command
        return sorted(array[s-1:e])[k-1]        
    return list(map(mapCommand,commands))

문제 해설

커맨드의 범위가 1부터이므로 실제 시작 인덱스는 -1 해줘야 한다.

0=<x <end 범위이므로 종료 인덱스는 그대로 쓰면 된다.

시작 인덱스와 같은 논리(포함)로 k인덱스는 -1 해준다.