프로그래머스|정렬|K번째수

문제설명

프로그래머스 정렬 연습의 K번째수 문제는 배열과 3가지 숫자를 입력받고 3가지 숫자에 맞추어 배열을 잘라내고 그 배열에서 k번째의 수를 출력하는 프로그램입니다.

​

작동 순서

1. 배열과 i, j, k 숫자가 담긴 commands를 입력받는다.

2. 배열의 i번째부터 j번째까지 잘라낸다.

3. 잘라낸 배열을 정렬한다.

4. 정렬된 배열에서 k번째수를 추출한다.

5. 이를 반복하고 완료되면 출력한다.

​

소스코드

def solution(array, commands):
    answer = []
    for i in range(len(commands)):
        extract=[]
        for j in range(commands[i][0]-1,commands[i][1]):
            extract.append(array[j])
        extract.sort()
        print(extract)
        answer.append(extract[commands[i][2]-1])
    return answer