알고리즘/프로그래머스

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

빙빙 2021. 7. 13. 08:43
def solution(array, commands):
    answer = []
    for i in range(len(commands)):
        start = commands[i][0] - 1
        end = commands[i][1]
        n = commands[i][2] - 1
        new = sorted(array[start:end])

        answer.append(new[n])

    return answer

단순하게 문제에 나와있는대로 풀었다!

command에서 시작점과 끝점,몇 번째 수를 받고

리스트 슬라이싱을 해서 정렬을 해줬다.

그리고 n번째 값을 answer에 append해줌!