[프로그래머스] 정렬 - K번째

 

 

내 답

def solution(array, commands):
    
    answer = []
    
    for c in commands:
        start = c[0]
        end = c[1]
        search = c[2]
        k = array[start-1:end]
        k.sort()
        answer.append(k[search-1])
    
    return answer

 

한 줄로 다 해보려다가 오류가 났다.

결국 풀어써서 성공..^^

.sort() 가 아니라 sorted() 이렇게 했어야 했나보다 ㅠ__ㅠ

 

다른 사람 답

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

 

인덱스로 저렇게 나눠서 받지 말고 그냥 리스트 개수에 맞게 변수 설정하는 것도 좋은듯!

잘 배워갑니다.