[BOJ] 두 배 더하기 12931번 C++

 

 

문제

모든 값이 0으로 채워져 있는 길이가 N인 배열 A가 있다. 영선이는 다음과 같은 두 연산을 수행할 수 있다.

• 배열에 있는 값 하나를 1 증가시킨다.
• 배열에 있는 모든 값을 두 배 시킨다.

배열 B가 주어졌을 때, 배열 A를 B로 만들기 위한 연산의 최소 횟수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 배열의 크기 N이 주어진다. (1 ≤ N ≤ 50)

둘째 줄에는 배열 B에 들어있는 원소가 공백으로 구분해서 주어진다. 배열에 B에 들어있는 값은 0보다 크거나 같고, 1,000보다 작거나 같다.

출력

첫째 줄에 배열 A를 B로 바꾸기 위한 최소 연산 횟수를 출력한다.

 

접근 방식

 

의외로 간단하게 풀 수 있다.

입력으로 받은 배열을 역으로 2로 나누거나 1을 빼서 0으로 만드는 횟수를 구하면된다.

이때, 만약 한 원소의 값이 홀수라면 1을 증가시킨것이기 때문에 카운트를 따로 세어주면 된다.

 

즉, 예를 들어 배열의 원소가 100일때

100 -> 50 -> 25 -> 24 -> 12 -> 6 -> 3 -> 2 -> 1 -> 0이러한 과정으로 9번이 걸리는 것을 알 수 있다.

 

 

코드

 

#include<bits/stdc++.h>
using namespace std;

bool check(const vector<int> & v) {
    for(auto c : v) if(c != 0) return false;
    return true;
}

int main() {
    int N, cnt = 0; cin>>N;
    vector<int> v(N);
    for(int i=0; i<N; i++) cin>>v[i];

    while(true) {
        
        for(int i=0; i<N; i++) {
            if(v[i] == 0) continue;
            int tmp = v[i];
            if(tmp % 2 == 1) {
                v[i] -= 1;
                cnt++;
            }
            v[i] /= 2;
        }

        if(check(v)) break;
        cnt++;
    }

    cout<<cnt;


}