BOJ-11497 통나무 건너뛰기 Swift

문제

남규는 통나무를 세워 놓고 건너뛰기를 좋아한다. 그래서 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)

출력

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

내가 푼 풀이

- 주어진 통나무를 오름차순으로 정렬한다.

- 정렬된 배열에서 각 원소는 자신과 인접한 원소가 원소값차이의 최소값이 된다.

- 주어진 통나무를 오름차순으로만 정렬한다면, 원소사이의 원소값차이가 최소값이 되지만 배열의 첫번째 원소와 마지막원소도 인접하므로 차이값은 최대가 된다.

- 배열의 원소중 최대값을 가운데 두고, 중앙을 기준으로 왼쪽과 오른쪽에 정렬된원소를 하나씩 넣는다면, 인접한 원소간 차이값이 최소가된다.

- 주어진 배열 [2,4,5,7,9] 로 예시를 들어보자

2

4

5

7

9

해당 배열은 이미 오름차순으로 정렬되있는 상태이다.

각 원소는 인접한 원소와 차이값이 최소로 되어있지만, 첫 번째 원소 2와 마지막원소 9의 차이는 7이므로 주어진 문제의 답과 거리가 멀다.

배열의 마지막원소가 배열의 최대값이므로 이를 중심으로 왼쪽과 오른쪽에 차이값이 적은 원소를 넣어주면 된다.

 

원소 9)

9

원소 7)

7

9

원소 5)

7

9

5

원소 4)

4

7

9

5

원소 2)

4

7

9

5

2

이렇게 정렬된 배열은 인접한 통나무의 높이 차가 최소가 된 배열이 된다.

해당 통나무 건너뛰기 난이도는 인접한 차이중 최대가 되는 9-5 = 4 가된다.

 

코드로 구현하면 다음과 같다.

import Foundation

// 테스트케이스 수
let c = Int(readLine()!)!

for i in 0..<c {
    // 주어진 통나무 수 와 높이 배열 입력
    let count = Int(readLine()!)!
    var arr = readLine()!.split(separator: " ").map{ Int($0)! }
    
    // 오름차순
    var sortedArr = arr.sorted{$0 < $1}
    
    // 배열 정렬
    var max = sortedArr.removeLast()
    var arr1 = [Int]()
    var arr2 = Array(repeating: 0, count: sortedArr.count / 2)
    var idx = 0
    while idx < sortedArr.count {
        if idx+1 == sortedArr.count {
            arr1.append(sortedArr[idx])
            idx += 1
        } else {
            arr1.append(sortedArr[idx])
            arr2[arr2.count-1-(idx/2)] = sortedArr[idx+1]
            idx += 2
        }
    }
    arr1.append(max)
    var r = arr1 + arr2
    
    // 인접한 통나무의 높이차이를 비교하여 가장 큰 차이를 저장
    var min = -1
    for j in 0..<r.count {
        var num = Int()
        if j == r.count - 1 {
            num = r[j] - r[0]
        } else {
            num = r[j] - r[j+1]
        }
        if min < abs(num) { min = abs(num) }
    }
    print(min)
}