본문 바로가기

알고리즘/프로그래머스

[프로그래머스 코딩테스트 고득점 Kit / 정렬] K번째수 (python)

728x90

풀이

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

 

commands 2차원 배열을 반복문을 돌면서 한 배열씩 꺼내서 i, j, k 변수에 저장하고,

array를 i부터 j번째 수까지 slice한 다음 sorted로 오름차순 정렬해준 뒤,

k번째 수를 answer에 저장한다.

반응형