[Algorithm] 이진 변환 반복하기

문제 설명

0과 1로 이루어진 어떤 문자열 x에 대한 이진 변환을 다음과 같이 정의합니다. 

• x의 모든 0을 제거합니다.
• x의 길이를 c라고 하면, x를 "c를 2진법으로 표현한 문자열"로 바꿉니다.

예를 들어, x = "0111010"이라면, x에 이진 변환을 가하면 x = "0111010" -> "1111" -> "100" 이 됩니다.

0과 1로 이루어진 문자열 s가 매개변수로 주어집니다. s가 "1"이 될 때까지 계속해서 s에 이진 변환을 가했을 때, 이진 변환의 횟수와 변환 과정에서 제거된 모든 0의 개수를 각각 배열에 담아 return 하도록 solution 함수를 완성해주세요.

 

문제 풀이

1. 문자열을 순회하며 0과 1의 개수를 구한다.
2. 0의 개수를 결과값에 더해준다.
3. 1의 개수로 새로운 바이너리 문자열을 만든다.
4. 문자열이 "1"이 될 때까지 1~3 과정을 반복한다.

 

코드 작성

#include <string>
#include <bitset>
#include <vector>

using namespace std;

std::string decimalToBinary(int number) {
    return std::bitset<32>(number).to_string(); // 32비트로 변환
}

vector<int> solution(string s) {
    vector<int> answer(2, 0);
    
    while (s != "1")
    {
        int zeroCount = 0;
        int oneCount = 0;
        
        // 0의 개수와 1의 개수를 구한다.
        for (const char& ch : s)
        {
            if (ch == '0') ++zeroCount;
            if (ch == '1') ++oneCount;
        }
        
        // 결과값을 계산한다.
        answer[0]++;
        answer[1] += zeroCount;
        
        // 1의 개수로 새로운 s를 만든다.
        string binaryString = decimalToBinary(oneCount);
        binaryString.erase(0, binaryString.find_first_not_of('0'));
        
        s = binaryString;
    }
    
    return answer;
}