[프로그래머스] K번째 수

kjh1551·2020년 9월 23일
0

알고리즘 풀이

목록 보기
9/14
post-thumbnail

문제

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.


def solution(array, commands):
    answer = []
    
    for com in commands:
        i = com[0]-1;
        j = com[1];
        k = com[2]-1;
        
        cut = array[i:j]
        cut.sort()
        answer.append(cut[k]);
        
    return answer

다른 풀이

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

i,j,k 를 한번에 넣을 수 있는 사실을 처음 알았다.
굳굳

0개의 댓글