[프로그래머스] 이진 변환 반복하기

문제 설명

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 함수를 완성해주세요.

제한사항

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

---

풀이

10분

0 제거 -> 길이를 이진수로 변환 -> "1"인지 확인

문제 설명에 나온 과정을 그대로 코드로 옮겼다.

코드

// 03:33 ~ 03:42
#include <string>
#include <vector>

using namespace std;

vector<int> solution(string s) {
    vector<int> answer;
    int loop_cnt=0, zero_cnt = 0;
    int n;
    string temp;
    while(s != "1"){
        temp = "";
        for(char c : s){
            if(c == '0') ++zero_cnt;
            else temp += c;
        }
        n = temp.length();
        temp = "";
        while(n > 0){
            if(n % 2 ==0) temp = '0' + temp;
            else temp = '1' + temp;
            n /= 2;
        }
        ++loop_cnt;
        s = temp;
    }
    answer.push_back(loop_cnt);
    answer.push_back(zero_cnt);
    return answer;
}