[정렬/탐색] 프로그래머스 K번째 수 - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

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

 

 

 

2. 문제 풀이

 주어진 배열의 일정 범위 내 K번째 수를 구하는 문제입니다.

 

sorted(lst): lst 내 원소를 오름차순으로 정렬

 파이썬의 sorted() 함수를 이용하면 쉽게 구할 수 있습니다.

 

 

 

3. 코드

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

 sorted() 함수를 사용해서 해당 범위의 배열을 오름차순 정렬합니다. 그리고 K번째 수를 뽑아냅니다.