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

출력

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

예제

Example 1:

Input: 
0001100
Output: 
1

조건

시간 제한 : 2초

메모리 제한 : 128 MB

풀이과정

내 풀이 1

import sys
S = str(sys.stdin.readline().rstrip())

count_0 = 0
count_1 = 0

li = []
index = 2
for i in S:
  if not li:
    li.append(i)
    if i == '0':
      count_0 += 1
    if i == '1':
      count_1 += 1
  else:
    if li[-1] != i and i == '0':
      count_0 += 1
    if li[-1] != i and i == '1':
      count_1 += 1
    li.append(i)

print(min(count_0,count_1))

우선 전체를 뒤집는 건 의미 없다는 걸 깨달았고 0이 연속인 것의 개수와 1이 연속인 것의 개수를 구해서 최솟값이 답이 되는걸 알아냈다. 하지만 역시나 머릿속에 있는걸 구현하는건 힘든일인가 보다.

내풀이 2

2달 가까이 지나서 다시 푼 내 풀이이다. 조금 더 간결해진거 같기도..하다.

s = input()

count_0 = 0
count_1 = 0
if s[0] == '0':
    count_0 += 1
    last = 0
else:
    count_1 += 1
    last = 1

for i in range(1, len(s)):
    if s[i] == '0':
        if last == 1:
            count_0 += 1
            last = 0
    else:
        if last == 0:
            count_1 += 1
            last = 1

print(min(count_0, count_1))

다른 풀이

다솜이는 모든 숫자를 전부 같게 만드는 것이 목적이다. 따라서 전부 0으로 바꾸는 경우와 전부 1로 바꾸는 경우 중에서 더 적은 횟수를 가지는 경우를 계산하면 된다.

data = input()
count0 = 0 # 전부 0으로 바꾸는 경우
count1 = 0 # 전부 1로 바꾸는 경우

# 첫 번째 원소에 대해서 처리
if data[0] == '1':
    count0 += 1
else:
    count1 += 1
    
# 두 번째 원소로부터 모든 원소를 확인하며
for i in range(len(data)-1):
    if data[i] != data[i+1]:
        # 다음 수에서 1로 바뀌는 경우
        if data[i+1] == '1':
            count0 += 1
        # 다음 수에서 0으로 바뀌는 경우
        else:
            count1 += 1
print(min(count0, count1))

기본적인 생각은 비슷한 것 같다. count0과 count1을 계산하는데 좀 더 깔끔한듯하다.