[Python] K번째수

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

---

풀이

1. 정렬하기 위해 sorted를 써주었다.

2. 인덱스는 0부터 시작하기때문에 i와 k를 사용할 때 각각 -1을 해주어야한다. 

 

---

def solution(array, commands):
    return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))

map과 lambda식을 사용하면 한줄로 표현할 수 있다.