호기심 많은 분석가

[프로그래머스] 정렬 - K번째수 (Python) 본문

Coding/Coding Test & Algorithm

[프로그래머스] 정렬 - K번째수 (Python)

DA Hun 2021. 6. 25. 14:36
 

코딩테스트 연습 - K번째수

[1, 5, 2, 6, 3, 7, 4] [[2, 5, 3], [4, 4, 1], [1, 7, 3]] [5, 6, 3]

programmers.co.kr

내 풀이

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

 다들 최대한 짧게 코딩하는 것을 좋아해 보여서 한 줄로 코딩해보았다. sort함수가 너무 잘 구현되어 있기 때문에 문제의 설명에 따라 진행하기만 하면 쉽게 해결할 수 있음. 또한 list comprehension은 역시 쓸 때마다 기분이 좋아지는 문법이야.