[Python] BOJ 1439번 - 뒤집기

문제 기술

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

예를 들어 S=0001100 일 때,

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

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

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

입력

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

출력

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

예제 입력 1

0001100

예제 출력 1

1

 

나의 코드

a = input()
b = list(map(int, list(str(a))))

count_0, count_1 = 0, 0

for i in range(len(b)):
    if i == 0:
        if b[i] == 0:
            count_0 += 1
        else:
            count_1 += 1 # 처음에 0인지 1인지에 따라 숫자가 바뀌는 횟수를 카운트
    else:
        if b[i-1] != b[i]:
            if b[i-1] == 0:
                count_1 += 1
            else:
                count_0 += 1

if count_0 > count_1:
    print(count_1)
else:
    print(count_0)

 # 이전 항과 다른데 전 항이 0이면 이제 1로 바뀐 것이므로 이 때에 count_1(1로 바뀌는 지점)을 1 더해줌
 # 반대로, 전 항이 1이면 이제 0이 된 것이므로 이 때에 count_0(0으로 바뀌는 지점)을 1 더해줌

0001100 일 경우, 0부터 시작하는 묶음은 2개, 1부터 시작하는 묶음은 1개이다. 이 경우 1부터 시작하는 한 개의 묶음만 뒤집어주면 모두 같은 숫자로 만들 수 있다.

이 문제를 풀 때 금방 풀지 못했던 이유는, 연속된 숫자의 갯수를 구하려고 했기 때문이다. 그냥 각 숫자로(1 또는 0) 뒤집어지는 횟수만 구하고(+ 처음에 시작한 묶음의 경우도 1을 추가해줘야 한다) 그 중 작은 숫자를 출력하면 되는 것이었다.

뒤집어야 하는 행동의 최소 횟수였기 때문에.. 문제를 잘 읽어야 한다고 생각했다.