Algorithm/Programmers

[Programmers] Lv1. K번째수

by somida 2021. 5. 12.

문제

바로가기

 

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

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

programmers.co.kr

 

풀이

  • slicing 기능을 통해 리스트를 자른 후 정렬
  • 그리고 해당 인덱스의 숫자를 리스트에 넣어 Return

 

코드

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

 

반응형

댓글