[C++] 백준 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로 바꿔주는 횟수, 모든 숫자를 0으로 바꿔주는 횟수를 세는 함수를 만들고 더 작은 값을 출력하도록 했다.

연속된 숫자를 뒤집는 것은 한번의 횟수로 세는 것을 기억하도록 하자.

그렇다면 모든 숫자를 1로 바꿔주는 경우는 0을 처음 만났을 때다. 즉 입력받은 문자열을 순서대로 탐색하다 0을 처음 만나게 되면 뒤집어주는 횟수를 증가시켜 주면 된다.

반대로 모든 숫자를 0으로 바꿔주는 경우는 1을 처음 만났을 때다. 즉 입력받은 문자열을 순서대로 탐색하다 1을 처음 만나게 되면 뒤집어주는 횟수를 증가시켜 주면 된다.

위에서 세어준 두 값중 더 작은 값을 출력하도록 한다.

 

코드

#include <iostream>
#include <string>
using namespace std;
 
// 모두 1로 바꾸는 것과 모두 0으로 바꾸는 것 두 가지 경우를 세서 작은 숫자를 출력한다.
 
// 1을 0으로 뒤집는 횟수
int zero(string n) {
    int size = n.size();
    int cnt = 0;
    bool z = true;    // 맨 앞에 1이 나와도 뒤집어줘야 해서 true로 초기화
    for (int i = 0; i < size; i++) {
        // 0이 나오면 나왔다는 표시로 z를 true로 바꿔줌.
        if (n[i] == '0') {
            z = true;
        }
        else {
            // 0이 이미 나왔고(z==true) 1을 만나면 뒤집어줘야 한다.
            // 다시 z는 false로 바꿔줌 (나오지 않았다고 가정해야 하기 때문)
            if (z == true) {
                cnt++;
                z = false;
            }
        }
    }
    return cnt;
}
 
// 0을 1로 뒤집는 횟수
int one(string n) {
    int size = n.size();
    int cnt = 0;
    bool o = true;    // 맨 앞에 0이 나와도 뒤집어줘야 해서 true로 초기화
    for (int i = 0; i < size; i++) {
        // 1이 나오면 나왔다는 표시로 o를 true로 바꿔줌.
        if (n[i] == '1') {
            o = true;
        }
        else {
            // 1이 이미 나왔고(o==true) 0을 만나면 뒤집어줘야 한다.
            // 다시 o는 false로 바꿔줌 (나오지 않았다고 가정해야 하기 때문)
            if (o == true) {
                cnt++;
                o = false;
            }
        }
    }
    return cnt;
}
 
int main() {
    string n;
    cin >> n;
 
    int result = zero(n) >= one(n) ? one(n) : zero(n);
    cout << result;
}

 

Memo

두 함수의 flag를 true로 설정해주지 않고 제출해서 아쉬운 문제였다.