[백준] 12931번. 두 배 더하기(JAVA)

문제

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

---

풀이(18분)

import java.io.*;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int n = Integer.parseInt(br.readLine());
        int[] arr = new int[n];
        boolean[] check = new boolean[n];
        int remaining = n;  // 아직 1이 아닌 수의 개수
        int answer = n;

        st = new StringTokenizer(br.readLine());
        for(int i = 0; i < n; i++){
            arr[i] = Integer.parseInt(st.nextToken());
            if(arr[i] == 1){
                check[i] = true;
                remaining--;
            }
            if(arr[i] == 0){
                check[i] = true;
                remaining--;
                answer--;
            }
        }

        loop1:
        while (remaining > 0) {
            for (int i = 0; i < n; i++) {
                if (check[i]) continue; // 이미 1인 경우 패스

                if (arr[i] == 1) {
                    check[i] = true; // 새로 1이된 경우 체크
                    remaining--;
                    if(remaining == 0){ // 남은 수가 없다면 바로 종료
                        break loop1;
                    }
                } else {
                    if (arr[i] % 2 == 1) { // 홀수인 경우
                        arr[i]--;
                        answer++;
                    }
                    arr[i] /= 2;
                }
            }
            answer++; // 모두 나눠진 한 턴
        }

        System.out.println(answer);
    }
}

---

문제 풀이 전략

 

배열 A -> B로 가는 것의 최소 횟수는 B -> A로 가는 것의 최소 카운트와 동일하다.

 

A -> B로 가는 경우 언제 0을 1로 바꿔줄 것인지가 조금 애매한 부분이 있었다.

따라서 B -> A로 가며, 홀수인 경우 무조건 -1을 해주고 전체 배열을 한 바퀴 돈 뒤에는 /= 2를 해준다.

최종적으로 해당 인덱스가 처음으로 1이 된 경우는 remaining를 -1 해준 뒤 끝까지 반복한다.

 

언제나 그렇지만 A -> B인 문제의 경우 B -> A가 편한 것인지를 잘 파악하는 게 중요하다고 느껴졌다.