취준/프로그래머스

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

puff 2020. 2. 9. 17:40

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

 

코딩테스트 연습 - K번째수 | 프로그래머스

[1, 5, 2, 6, 3, 7, 4] [[2, 5, 3], [4, 4, 1], [1, 7, 3]] [5, 6, 3]

programmers.co.kr

 

설명이 필요할거 같지는 않다.

명령어 단위로 잘라서 정렬하고 위치를 answer에 저장한다.

프로그래머스 문제들은 답은 원소 위치를 1부터 시작하게 주고 정작 데이터는 0부터 시작하게 주는 경우가 좀 있다.

리스트 인덱스에 조심하자.

def solution(array, commands):
    answer = []
    for c in commands:
        a = sorted(array[c[0]-1:c[1]])
        answer.append(a[c[2]-1])

    return answer