k 번째 수

문제 설명

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.

핵심아이디어

첫 번째 인덱스는 0으로 시작하기 때문에, 슬라이싱을 i-1로 두면 편하다.

sorted() 함수를 사용하면 리스트를 오름차순으로 정렬할 수 있다.

첫 코드

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

개선된 코드

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