[프로그래머스]K번째수-정렬
SW개발/코딩테스트

[프로그래머스]K번째수-정렬

문제 설명

 

def solution(array, commands):
    answer = []

    for i in commands:
        temp = array[i[0]-1:i[1]]
        temp.sort()
        answer.append(temp[i[2]-1])

    return answer

 

코드 설명

commands 리스트 안에서 for 반복문을 돌려 commands[i][0:2] 에 해당하는 원소들을 리스트로서 접근하여 temp 리스트에

(i ~ j 번째를 자른 결과) 넣어준다. 그 후 정렬을 하고 정렬된 리스트의 i[2]-1 인덱스(k를 의미함)에 접근하여 answer 리스트에 원소를 추가하여 준다.

 

Point : 파이썬에는 내장 정렬 함수인 sort(), sorted()가 존재하기에 이를 활용하면 정렬을 위한 코드를 작성하지 않아도 된다. 또한 리스트 안에서 for 반복문을 사용할 수 있으므로 보다 리스트에 쉽게 접근하여 사용할 수 있다.

728x90