[Algorithm] [Python&Swift] BOJ/백준 - 1439 _ 뒤집기

1439 - 뒤집기

 

문제

---

다솜이는 0과 1로만 이루어진 문자열 S를 가지고 있다. 다솜이는 이 문자열 S에 있는 모든 숫자를 전부 같게 만들려고 한다. 다솜이가 할 수 있는 행동은 S에서 연속된 하나 이상의 숫자를 잡고 모두 뒤집는 것이다. 뒤집는 것은 1을 0으로, 0을 1로 바꾸는 것을 의미한다.

예를 들어 S=0001100 일 때,

1. 전체를 뒤집으면 1110011이 된다.
2. 4번째 문자부터 5번째 문자까지 뒤집으면 1111111이 되어서 2번 만에 모두 같은 숫자로 만들 수 있다.
3. 하지만, 처음부터 4번째 문자부터 5번째 문자까지 문자를 뒤집으면 한 번에 0000000이 되어서 1번 만에 모두 같은 숫자로 만들 수 있다.

문자열 S가 주어졌을 때, 다솜이가 해야하는 행동의 최소 횟수를 출력하시오.

 

입력

---

첫째 줄에 문자열 S가 주어진다. S의 길이는 100만보다 작다.

 

출력

---

첫째 줄에 다솜이가 해야하는 행동의 최소 횟수를 출력한다.

 

제출코드

---

Python

arr = list(map(int, input()))
count0 = 0  # 모든 숫자를 0으로 만들기 위해 필요한 연산 횟수
count1 = 0  # 모든 숫자를 1로 만들기 위해 필요한 연산 횟수

if arr[0] == 1:
    count0 += 1
else:
    count1 = 0

for i in range(len(arr)-1):
    if arr[i] != arr[i+1]:
        # 0 -> 1로 바뀌는 경우, 따라서 해당 1을 0으로 바꿔줘야 하기 때문에 count0 증가
        if arr[i+1] == 1:
            count0 += 1
        # 1 -> 0으로 바뀌는 경우, 따라서 해당 0을 1로 바꿔줘야 하기 때문에 count1 증가
        else:
            count1 += 1
print(min(count0, count1))

 

Swift

let arr: String = readLine()!
var zeroCount = 0
var oneCount = 0
var before = arr.first!
if before == "0" {
    oneCount += 1
} else {
    zeroCount += 1
}
for num in arr {
    if num != before {
        if num == "1" { zeroCount += 1 }
        else { oneCount += 1 }
        before = num
    }
}
print(min(zeroCount, oneCount))

문제풀이

---

주어진 문자열에서 최소한의 행동 을 통해 모두 같은 숫자로 만들어내야 한다.

 

주어진 문자열의 0과 1의 빈도에 따라 모두 0으로 만들거냐 혹은 모두 1로 만들거냐가 결정나기 때문에 두 가지 케이스에 대한 모든 케이스를 구해야했다.

 

따라서, 현재 주어진 문자열을 순회하며 문자가 변경되는 경우를 통해 필요한 연산 횟수를 계산한다.