[프로그래머스/Python][정렬]K번째 수

코딩테스트 연습 - K번째수

---

문제

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

예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면

1. array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다.
2. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다.
3. 2에서 나온 배열의 3번째 숫자는 5입니다.

배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

풀이

 파이썬으로 알고리즘 문제를 풀어본 경험이 없어 파이썬스럽지 않게 풀이했다. 추후에 파이썬 리스트에 대해 정리해서 블로그 글 올릴 예정.

 

배열 array와 i,j,k가 저장되어있는 이차원 배열 commands를 받는 함수를 만든다.

comands의 length만큼 반복하는 for문을 만들고(k번재 수를 구할 횟수), i~j수가 들어있는 배열을 생성한다 -> 나는 배열 생성 후 for문으로 넣었는데, 이제 보니 lambda를 사용하는게 더 나을 것 같다.

배열을 정렬하고, k번째 수를 뽑아 정답 배열에 넣는다.

소스코드

def solution(array, commands):
    answer = [0] * len(commands)
    # an =[] 공간 설정을 할 필요 없어서 이렇게 해도 됨 -> 오류가 나는 이유는 할당을 해서. append를 해야 오류가 안남.
    # an = list() 이렇게 해도 됨
    cnt = 0

    for c in commands:  # comands 개수만큼 반복
        arr = [0] * (c[1]-c[0]+1)  # [i,j,k]에서 j-i+1길이의 배열새엇ㅇ

        i_cnt = 0
        for i in range(c[0]-1, c[1]):  # i~j까지 증가, arr에 넣음
            arr[i_cnt] = array[i]
            i_cnt += 1
            # .append() 사용

        arr.sort()  # arr 정렬

        answer[cnt] = arr[c[2]-1]  # 정답 배열(answer)에 k번째 수 넣기
        cnt += 1

    return answer

이렇게 해도 풀리긴 했지만...