프로그래머스

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

hotchya 2021. 9. 8. 20:45

K번째수

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

문제 단순화

  • 정수배열 array, [i,j,k]를 요소로 가진 command 배열이 있다.
  • 각 command 요소에 대해서 array의 i번째에서 j번째 사이의 k번째로 작은 수를 구한다.

문제풀이

  • 각 command 요소마다 array를 자르고 정렬한다.
  • 정렬된 자른 배열의 k번째 수를 모아서 리턴한다.

  • 코드 실행시간이 넉넉하다.

코드

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