파이썬 그리디 알고리즘 백준 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만보다 작다.

출력

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

풀이

같은 숫자로 만들기 위해 최소 횟수로 뒤집으려면 연속하는 0의 개수와 연속하는 1의 개수를 구하여

더 작은 횟수를 뒤집으면 된다.

연속하는 0과 1의 개수를 구하기 위해

처음 state를 첫번째 문자로 설정하고 문자열을 인덱스 1부터 순회하여

state와 문자가 다르면 state상태에 따라 zero와 one를 1씩 더한다.

이렇게 하면 가장 마지막 연속하는 수를 셀 수 없기 때문에 문자열의 마지막 인덱스에 따라

다시 zero와 one를 1씩 더한다.

그리고 zero와 one중 작은 값을 뒤집으면 같은 숫자로 문자열을 만들 수 있다.

 

import sys
strings = sys.stdin.readline().rstrip("\n")
zero = 0
one = 0
state = strings[0]
for i in range(1,len(strings)):
    if state == strings[i]:
        continue
    elif state != strings[i] and state == "0":
        zero +=1
        state = strings[i]
    elif state != strings[i] and state == "1":
        one +=1
        state = strings[i]
if strings[len(strings)-1] == "0":
    zero += 1
elif strings[len(strings)-1] == "1":
    one += 1

print(min(zero, one))