[프로그래머스] lv1 K번째수 / 파이썬, 고득점kit

K번째수

파이썬으로 풀 경우 리스트 인덱싱의 의미를 알아야 한다.

주의점!

• 인덱스 넘버는 0부터 시작한다는 것을 명심하자!

 

문재해결 아이디어

• 특정 구간만큼 array의 값을 입력받아줄 새로운 리스트를 생성한다
• 입력받은 commands의 수 만큼 반복한다

 

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

print(solution([1, 5, 2, 6, 3, 7, 4], [[2, 5, 3], [4, 4, 1], [1, 7, 3]]	))