[프로그래머스] K번째 수(파이썬)

코딩테스트 연습 - K번째수 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

[문제 요약]

- 입력으로 리스트 2개가 주어진다.

- commands 리스트의 첫번째 인자부터 두번째 인자까지 추출한 뒤 세번째 인자 위치의 수를 출력한다.

- commands 리스트의 길이만큼 반복한다.

 

[문제 풀이 방법]

- 2번째부터 5번째까지 자르면 -> 인덱스로는 1번 ~ 4번이기 때문에 처음 범위에서 1을 빼줬다.

- 리스트를 오름차순으로 정렬하고 또 다른 리스트에 3번째 숫자를 추가한다.

 

[정답]

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