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

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

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

배열 array의 i번째 숫자부터 j번째 숫자까지 고르고 정렬한 후 k번째 수를 구하는 문제이다.

입력 형태는 아래와 같다.

기본적인 solution()함수의 매개변수로 array,commands를 받게되는 것이다. commands안에 각각의 상황에 따른 i,j,k값이 있고, 각 command별로 결과값들을 모아서 리스트로 반환하는 것이다. 아래와 같이 구현하였다

from typing import MutableSequence

def solution(array, commands) ->MutableSequence:
    answer = []
    for a in commands:
        i,j,k = a
        answer.append(sorted(array[i-1:j])[k-1])
    return answer