[Gold II] Title: 색깔 통일하기, Time: 76 ms, Memory: 101448 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/17307. 색깔 통일하기/README.md

@@ -0,0 +1,54 @@
+# [Gold II] 색깔 통일하기 - 17307 
+
+[문제 링크](https://www.acmicpc.net/problem/17307) 
+
+### 성능 요약
+
+메모리: 101448 KB, 시간: 76 ms
+
+### 분류
+
+누적 합
+
+### 제출 일자
+
+2026년 1월 11일 18:02:31
+
+### 문제 설명
+
+<p><em>N</em>개의 버튼이 일렬로 나열되어 있다. 이 버튼들은 바로 양옆에 인접한 버튼끼리 연결되어 있다. 각 버튼은 LED가 내장되어있어 총 <em>C </em>종류의 색을 띨 수 있다. 그 색깔들을 각각 0번 색깔, 1번 색깔, ..., (<em>C</em>-1)번 색깔이라고 하자.</p>
+
+<p>이때 현재 색깔이 <em>x</em>인 한 버튼을 누르면 누른 버튼과 함께 해당 버튼의 양옆으로 같은 색으로 연속해서 이어진 모든 버튼이 (<em>x</em>+1)%<em>C</em> 번 색깔로 변한다. 우리의 목표는 버튼을 누르는 횟수를 최소화하며 모든 버튼을 같은 색깔로 만드는 것이다.</p>
+
+<p>예를 들어, <em>N</em>=5, <em>C</em>=4일 때 아래와 같은 색의 버튼들이 있다고 하자.</p>
+
+<p style="text-align: center;"><img alt="" src="https://upload.acmicpc.net/6a34baca-47f5-4d6b-b5e9-a5dc5e573f78/-/preview/" style="height: 109px; width: 600px;"></p>
+
+<p>여기서 4번 버튼을 누르면 현재 4번 버튼의 양옆에 같은 색의 버튼이 없으므로 4번 버튼의 색만 0번 색깔로 바뀌게 된다.</p>
+
+<p style="text-align: center;"><img alt="" src="https://upload.acmicpc.net/3ff216fc-c676-4999-b350-d58c6d13a2a8/-/preview/" style="width: 600px; height: 109px;"></p>
+
+<p>그 이후 2번째 버튼을 누르면 2번의 왼쪽으로는 같은 색의 버튼이 없고, 오른쪽으로 연속한 3, 4번 버튼이 2번 버튼의 색과 같으므로 2, 3, 4번 버튼의 색이 1번 색깔로 바뀐다.</p>
+
+<p style="text-align: center;"><img alt="" src="https://upload.acmicpc.net/3606753a-6a72-49ad-b1ca-aab5c078f9ab/-/preview/" style="height: 109px; width: 600px;"></p>
+
+<p>그 이후 3번째 버튼을 누르면 1, 2, 3, 4, 5번 버튼의 색이 모두 함께 2번 색깔로 바뀌게 된다.</p>
+
+<p style="text-align: center;"><img alt="" src="https://upload.acmicpc.net/697d01d9-a812-4842-b3eb-c0393e7e8e11/-/preview/" style="height: 109px; width: 600px;"></p>
+
+<p>우리의 목적은 버튼을 누르는 횟수를 최소화 하면서 모든 버튼의 색을 한 색으로 통일시키는 것이다. 위의 방법대로면 4번 버튼을 누른 후 2번 버튼을 누르면서 2회만에 1번 색으로 통일된다.</p>
+
+<p>하지만 지금은 어떤 이유로 오직 하나의 버튼만 누를 수 있기에, 4번 버튼을 누르고 2번 버튼을 누르는 방법은 쓸 수 없게 되었다. 그렇다면 어떤 버튼을 선택해야 해당 버튼을 최소한으로 누르며 모든 버튼의 색을 통일시킬 수 있을까?</p>
+
+### 입력 
+
+ <p>첫 번째 줄에 버튼의 수 <em>N</em>(1 ≤ <em>N</em> ≤ 250,000)과 가능한 색의 수 (1 ≤ <em>C</em> ≤ 10<sup>9</sup>)가 공백으로 구분되어 주어진다.</p>
+
+<p>다음 줄에 현재 각 버튼의 색 <em>X<sub>i</sub></em> (0 ≤<em> X<sub>i</sub></em> < <em>C</em>, 1 ≤ <em>i</em> ≤ <em>N</em>)가 공백으로 구분되어 주어진다.</p>
+
+### 출력 
+
+ <p>첫 번째 줄에 몇 번 버튼을 눌러야 하는지 출력한다, 버튼에는 가장 왼쪽에서 시작해 1번부터 <em>N</em>번까지 번호가 차례로 매겨져 있다.</p>
+
+<p>두 번째 줄에 모든 버튼을 같은 색으로 통일시키기 위해 그 버튼을 눌러야 할 횟수를 출력한다. <strong>만약 최소 횟수가 되는 버튼이 여러 개 존재한다면 그중 가장 왼쪽의 버튼을 출력한다.</strong></p>
+

백준/Gold/17307. 색깔 통일하기/색깔 통일하기.swift

@@ -0,0 +1,95 @@
+import Foundation
+
+final class FastIO {
+    private var data: [UInt8] = Array(FileHandle.standardInput.readDataToEndOfFile()) + [0]
+    private var idx: Int = 0
+    
+    func readInt() -> Int {
+        var num = 0
+        var sign = 1
+        
+        while idx < data.count && (data[idx] == 10 || data[idx] == 13 || data[idx] == 32) {
+            idx += 1
+        }
+        
+        if idx < data.count && data[idx] == 45 {
+            sign = -1
+            idx += 1
+        }
+        
+        while idx < data.count && data[idx] >= 48 {
+            num = num * 10 + Int(data[idx] - 48)
+            idx += 1
+        }
+        
+        return num * sign
+    }
+}
+
+let io = FastIO()
+
+let N = io.readInt()
+let C = io.readInt() - 1
+
+if N == 1 {
+    _ = io.readInt()
+    print(1)
+    print(0)
+    exit(0)
+}
+
+var button = Array(repeating: 0, count: N + 2)
+for i in 1...N {
+    button[i] = io.readInt()
+}
+
+var sum = Array(repeating: [Int64](repeating: 0, count: 2), count: N + 2)
+
+// 오른쪽 누적합
+for i in 2...N {
+    let prev = button[i - 1]
+    let cur = button[i]
+    
+    let cost: Int
+    if prev <= cur {
+        cost = cur - prev
+    } else {
+        cost = (C - prev) + cur + 1
+    }
+    
+    sum[i][0] = sum[i - 1][0] + Int64(cost)
+}
+
+// 왼쪽 누적합
+var i = N - 1
+while i >= 1 {
+    let next = button[i + 1]
+    let cur = button[i]
+    
+    let cost: Int
+    if next <= cur {
+        cost = cur - next
+    } else {
+        cost = (C - next) + cur + 1
+    }
+    
+    sum[i][1] = sum[i + 1][1] + Int64(cost)
+    i -= 1
+}
+
+var bestCost = Int64.max
+var bestIndex = 1
+
+for i in 1...N {
+    let right = sum[N][0] - sum[i][0]
+    let left = sum[1][1] - sum[i][1]
+    let cur = max(right, left)
+    
+    if cur < bestCost {
+        bestCost = cur
+        bestIndex = i
+    }
+}
+
+print(bestIndex)
+print(bestCost)