[Python] 백준 1439번 - 뒤집기

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

1439번: 뒤집기

 

혼자 힘으로 풀었는가? O

알고리즘 분류
 - 문자열
 - 그리디 알고리즘

 

문제

다솜이는 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로 나누어서 가장 적은쪽의 개수를 출력하였다.

 

import sys
input = sys.stdin.readline

s = input().rstrip()

if '0' not in s:
    print(0)
elif '1' not in s:
    print(0)
else:
    s0 = s.split('1')
    s1 = s.split('0')
    result0 = []
    result1 = []
    for i in range(len(s0)):
        if s0[i] != '':
            result0.append(s0[i])
    for i in range(len(s1)):
        if s1[i] != '':
            result1.append(s1[i])
            
    print(min(len(result0), len(result1)))

 

근데 이보다 더 간단한 방법이 있었다.

 

01 과 10의 숫자가 바뀌는 분기점을 세는방법이다.

 

import sys
input = sys.stdin.readline

a = input().rstrip()
print(max(a.count('01'), a.count('10')))