프로그래머스 Level 1 | K번째 수 | Python

tomkitcount·2025년 10월 3일

알고리즘

목록 보기
195/304

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


문제 파악

배열 array, [i,j,k] 를 입력 받았을 때,
배열을 i번째부터 j번째까지 자르고 오름차순으로 정렬한 뒤, k번째 수를 구하는 문제입니다.

해답 및 풀이

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

입력받은 commands에 대해 하나씩 순회하며 i,j,k를 할당해주고
array를 적절히 슬라이싱하여 new_array를 만든 후, 정렬한뒤
문제에서 요구하는 k번째 숫자를 answer list에 appned한 후 return 해주면 되는
비교적 간단한 문제였습니다.

profile
To make it count

0개의 댓글