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

문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/42748

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

문제에 나와있는대로 코드를 작성했다.

 

부분 배열의 시작, 끝 인덱스가 주어지면 부분 배열을 얻어내고 이를 정렬하여 부분 배열의 K번째 수를 얻는 과정을 반복한다.

def solution(array, commands):
    answer = []
    
    for command in commands:
        i, j, k = command[0], command[1], command[2]
        
        part_arr = array[i-1:j]
        part_arr.sort()
        answer.append(part_arr[k-1])
    
    return answer