프로그래머스 - 이진 변환 반복하기(C++)

문제 설명

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

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

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

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

---

제한사항

• s의 길이는 1 이상 150,000 이하입니다.
• s에는 '1'이 최소 하나 이상 포함되어 있습니다.

---

입출력 예

 

입출력 예 설명

입출력 예 #1

• 110010101001이 1이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

• 3번의 이진 변환을 하는 동안 8개의 0을 제거했으므로, [3,8]을 return 해야 합니다.

입출력 예 #2

• 01110이 1이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

• 3번의 이진 변환을 하는 동안 3개의 0을 제거했으므로, [3,3]을 return 해야 합니다.

입출력 예 #3

• 1111111이 1이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

• 4번의 이진 변환을 하는 동안 1개의 0을 제거했으므로, [4,1]을 return 해야 합니다.

 

풀이

#include <string>
#include <vector>

using namespace std;

int getOneCount(string s) {
    int oneCount = 0;
    
    for (int i = 0; i < s.size(); i++) {
        if (s[i] == '1') ++oneCount;
    }
    
    return oneCount;
}

string transformToBinary(int decimal) {
    string binary;
    
    while (decimal != 0) {
        binary = to_string(decimal % 2) + binary;
        decimal /= 2;
    }
    
    return binary;
}

vector<int> solution(string s) {
    vector<int> answer(2);
    
    do {
        int oneCount = getOneCount(s);
        ++answer[0];
        answer[1] += s.size() - oneCount;
        s = transformToBinary(oneCount);
    } while (s != "1");
    
    return answer;
}