[백준 1439] 뒤집기 (Feat.Java)

문제

다솜이는 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 java.util.*;
public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        String input = sc.nextLine();
        List<Integer> zeros = new ArrayList<>();
        List<Integer> ones = new ArrayList<>();
        char first = input.charAt(0);
        int count =0;

        for(int i = 1; i < input.length(); i++){
            count++;
            if(first != input.charAt(i)){
                if(first == '0') zeros.add(count);
                else ones.add(count);
                first =  input.charAt(i);
                count = 1;
            }
            if(i == input.length()-1){
                if(first == '0') zeros.add(count);
                else ones.add(count);
            }
        }// end for



        System.out.println(zeros.size() > ones.size() ? ones.size() : zeros.size());
    }
}

• 위 코드를 보면 List로 zeros, ones 를 선언
  • String으로 입력받은 input을 순회하면서 0 -> 1로 바뀌거나 1 -> 0으로 바뀌게 되었을 때 List에 count한 값들을 추가
• for문안에서 문자열의 마지막 char를 선택하고 있을 때, 그때까지의 count를 넣는 로직을 넣어주었다.
• 마지막 3항 연산자를 통해 size() 가 더 작은 값을 출력하게 하였다.