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

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

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

python 라이브러리의 sort를 사용해서 풀면 되는 문제입니다.

 

문제부터 요약하면, 

배열과 명령들이 주어지는데 각 명령 [i, j, k]는 array의 i~j번째 까지의 원소중 k번째를 요구합니다. 이렇게 각각의 요구사항들을 list에 넣어서 return 해야하는 문제입니다. 

 

<Solution> 

각각의 command에 대해 답을 구한후 append 해서 return 시키면 되는 간단한 문제입니다. 

실제 구현은 아래와 같습니다.

def solution(array, commands):
    def get_answer(array, command):
        temp_list = array[command[0]-1:command[1]]
        s = list(sorted(temp_list))
        return s[command[2]-1]

    answer = []
    for command in commands:
        answer.append(get_answer(array,command))

    return answer