[백준] 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만보다 작다.

출력

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

내 풀이

def convert01(str1):
    a = str1.count('01')
    b = str1.count('10')
    if a == 0 and b == 0:
        print(0)
    elif a == 0 and b != 0:
        print(b)
    elif a != 0 and b == 0:
        print(a)
    else:         
        if a > b:
            print(b)
        else:   # a <= b
            print(a)
        
convert01(input())

틀렸습니다! ㅠ

코멘트

처음에는 0과 1을 어떻게 분리해서 이 갯수를 찾을까 고민하고 있었는데 대상혁님이 01 혹은 10의 갯수를 세면 된다는 힌트를 줘서 풀어봤는데도 틀렸다. 예시 입출력에서는 전부 정답을 출력하는데 뭐를 잘못했을까...

References

https://www.acmicpc.net/problem/1439