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

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

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

문제 설명

· 숫자 배열 array와 [i, j, k]의 command로 이루어진 commands가 주어짐

· 각 command에 대해 array를 i에서 j까지 잘라서 정렬했을 때, k번째 수 반환

· 각 command의 결과로 이루어진 배열 반환

주어진 command에 대해 배열을 자르고 정렬하는 문제

접근 방법

· commands에 대해 List Comprehension을 통해 각 command를 받아 온 후 슬라이싱을 통해 자르고, 정렬한 후 k-1번째 인덱스 참조

알고리즘 설계 및 구현

def solution(array, commands):
    return [sorted(array[i-1:j])[k-1] for i, j, k in commands]

결과