Algorithm/프로그래머스

[프로그래머스/Python] K번째수 - Level1

poppy 2021. 5. 9. 14:06
반응형

programmers.co.kr/learn/courses/30/lessons/42748?language=python3

 

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

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

programmers.co.kr

def solution(array, commands):
    answer = []
    
    for command in commands:
        result = array[command[0]-1:command[1]] # i번째부터 j번째 숫자까지 자르기
        result.sort() # 자른 배열 정렬하기
        answer.append(result[command[2]-1]) # k번째 수 구하기
    
    return answer

먼저 입력 받은 배열에서 i번째 숫자부터 j번째 숫자까지 배열을 자른 후 result 에 저장합니다. 그 다음 자른 배열인 result를 정렬합니다. 마지막으로 정렬 후 result에서 k번째 숫자를 구한 후 answer 배열에 추가합니다.

반응형