[Python] 뒤집기 (백준 1439번 파이썬)

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

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

예제 입력 2

11111

예제 출력 2

0

예제 입력 3

00000001

예제 출력 3

1

예제 입력 4

11001100110011000001

예제 출력 4

4

예제 입력 5

11101101

예제 출력 5

2

 

 

0과 1로 이루어진 문자열에서 한 개의 숫자로 바꾸는데 걸리는 횟수를 묻는 문제입니다.

문제 이해 자체는 금방 했고 아이디어도 금방 떠올렸는데 구현에 있어서 시간을 너무 많이 소모했습니다(쩝...)

 

 

while문 내에서 for문을 통해 문자열을 검사합니다.

이때, 앞글자와 현재 글자가 다르면 현재글자의 이전 인덱스까지만 배열에 저장하고, 현재 인덱스부터 새로 문자열을 설정합니다.

그리고 while문의 종료 조건을 설정하는게 조금 까다로웠는데, 결과를 01, 00, 11, 10 이렇게 모두 나눠서 일일히 시도해보면서 고쳤습니다(ㅠㅠ)

 

연속된 문자를 끊어서 배열에 저장하고 나면, 0과 1 중에서 어떤게 더 많이 연속되어 있는지를 판단합니다.

0으로만 된 것과 1로만 된 것의 갯수를 세서 큰 것과 작은 것 중에서 작은 것의 갯수를 출력하면 결과가 됩니다.

 

코드

#백준 1439 뒤집기

s = input()
before = s[0]
arr = []
i = 1
while s:
    for i in range(1,len(s)):
        if s[i] != s[i-1]: #두개가 다르면
            arr.append(s[:i])
            s=s[i:]
            if len(s) == 1:
                arr.append(s[-1])
                s = ''
            break
    else:
        arr.append(s)
        s=''

zero, one = 0,0
for j in arr:
    if j[0] == '0':
        zero += 1
    else:
        one += 1
if one >= zero:
    print(zero)
else:
    print(one)