[Level 1] [python] K번째수

문제

파이썬 풀이

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

배열을 인덱싱을 이용하면 쉽게 풀 수 있는 문제였다.

i번째는 인덱스로 i-1이고 인덱싱에서 end는 포함하지 않기 때문에 (j-1) 번째에 +1을 해줘서 j까지로 잘라낸다.

그리고 파이썬의 sort()함수를 이용해 잘라낸 배열을 정렬해주었다.

그리고 k번째(인덱스 k-1) 값을 answer배열에 추가해주었다.