[프로그래머스] k번째 수 - python

문제

• 프로그래머스 k번째 수
• https://programmers.co.kr/learn/courses/30/lessons/42748

풀이

문제에서 요구한대로 배열을 자르고 오름차순 정렬 후 k번째 수를 구하면 된다.

i의 0번째 값 -> 시작 index

i의 1번째 값 -> 끝 index

i의 2번째 값 -> k번째 수

따라서 array[시작 index-1 : 끝 index]로 문자열을 자르고 정렬 후 k번째 수를 구하면된다.

코드

def solution(array, com):
    answer = []
    arr = []
    for i in com:
        arr=array[i[0]-1:i[1]]
        arr.sort()
        answer.append(arr[i[2]-1])
    return answer