파이썬 그리디 알고리즘 백준 11497 통나무 건너뛰기

문제

남규는 통나무를 세워 놓고 건너뛰기를 좋아한다. 그래서 N개의 통나무를 원형으로 세워 놓고 뛰어놀려고 한다. 남규는 원형으로 인접한 옆 통나무로 건너뛰는데, 이때 각 인접한 통나무의 높이 차가 최소가 되게 하려 한다.

통나무 건너뛰기의 난이도는 인접한 두 통나무 간의 높이의 차의 최댓값으로 결정된다. 높이가 {2, 4, 5, 7, 9}인 통나무들을 세우려 한다고 가정하자. 이를 [2, 9, 7, 4, 5]의 순서로 세웠다면, 가장 첫 통나무와 가장 마지막 통나무 역시 인접해 있다. 즉, 높이가 2인 것과 높이가 5인 것도 서로 인접해 있다. 배열 [2, 9, 7, 4, 5]의 난이도는 |2-9| = 7이다. 우리는 더 나은 배열 [2, 5, 9, 7, 4]를 만들 수 있으며 이 배열의 난이도는 |5-9| = 4이다. 이 배열보다 난이도가 낮은 배열은 만들 수 없으므로 이 배열이 남규가 찾는 답이 된다.

입력

입력은 T개의 테스트 케이스로 이루어져 있다. 첫 줄에 T가 주어진다.

이어지는 각 줄마다 첫 줄에 통나무의 개수를 나타내는 정수 N(5 ≤ N ≤ 10,000), 둘째 줄에 각 통나무의 높이를 나타내는 정수 Li가 주어진다. (1 ≤ Li ≤ 100,000)

출력

각 테스트 케이스마다 한 줄에 주어진 통나무들로 만들 수 있는 최소 난이도를 출력하시오.

풀이

통나무의 난이도는 인접하는 통나무 높이의 차다. 또한 통나무는 원형으로 구성되

입력받는 첫 값과 마지막 값도 인접해 있는 것이다.

이 문제를 해결하기 위해 입력값을 heapq 리스트로 생성하고 pop을 통해 가장 작은 값을 가져온다.

그리고 pop한 값을 새로운 리스트의 맨 뒤에 삽입하고, 다음 pop한 값을 리스트의 가장 앞에 삽입했다.

heapq가 남지 않을 때 까지 반복한다.

예로 살펴보면,

2 4 5 7 9 라는 입력값이 있고, temp 리스트는 비어있다.

처음 pop한 값인 2를 가장 뒤에 삽입한다. temp=[2]

다음 pop한 값인 4를 맨 앞에 삽입한다. temp=[4,2]

반복한다. temp=[7,4,2,5,9]

이것이 통나무를 건너 뛸 때의 최소 난이도이고, 난이도가 몇인지 구해야 하기 때문에

반복문을 통해 인접하는 두 수의 차의 절대값이 가장 큰 수를 구하고 출력한다.

 

import sys
import heapq
testCases = int(sys.stdin.readline().rstrip("\n"))
for i in range(testCases):
    n = int(sys.stdin.readline().rstrip("\n"))
    nheaps = list(map(int, sys.stdin.readline().rstrip("\n").split()))
    heapq.heapify(nheaps)
    temp = []
    while nheaps:
        temp.append(heapq.heappop(nheaps))
        if(nheaps):
            temp.insert(0,heapq.heappop(nheaps))

    max=-1
    for i in range(0, len(temp)):
        if(i+1<len(temp)):
            if(abs(temp[i]-temp[i+1])>max):
                max = abs(temp[i]-temp[i+1])
        else:
            if(abs(temp[len(temp)-1]-temp[0])>max):
                max = abs(temp[len(temp)-1]-temp[0])
    print(max)