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

* 문제

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

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

 

* 문제 풀이 

- 첫째, 반복문을 사용해 commands 2차원 리스트에서 순서대로 값을 불러온다.

- 둘째, array 리스트에서 slicing을 사용해 값을 추출한다.

- 셋째, sort 함수를 사용해 정렬한 후 K번째 값을 추출한다.

 

* 생각

- 간단한 문제였지만 sort 함수를 사용하면 문제가 될까? 걱정하기는 했다.

- 다른 사람의 풀이는 람다함수를 사용해서 한줄로 푼 것도 있는데 대단하다.

 

* 코드 

def solution(array, commands):
    answer_list = []
    for command in commands:
        temp_list = array[command[0]-1:command[1]]
        temp_list.sort()
        answer_list.append(temp_list[command[2]-1])
    
    return answer_list