[프로그래머스] 이진 변환 반복하기 / 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'이 최소 하나 이상 포함되어 있습니다.

 

입출력 예

s	result
"110010101001"	[3,8]
"01110"	[3,3]
"1111111"	[4,1]

 

 

코드

#include <string>
#include <vector>

using namespace std;

vector<int> solution(string s) {
    vector<int> answer;
    string one = "";
    int i;
    int cnt = 1;
    int zero = 0;
    int len = s.size();
    
    for (i = 0; i < s.size(); i++)
    {
        if (s[i] == '1')
            one += s[i];
        else
            zero++;
    }
    len = one.size();
    one.clear();
    while (len != 1)
    {
        cnt++;
        while (len)
        {
            if (len % 2)
                one += "1";
            else
                zero++;
            len /= 2;
        }
        len = one.size();
        one.clear();
    }
    answer.push_back(cnt);
    answer.push_back(zero);
    return answer;
}

 

우선 s를 대상으로 0을 제거했다. s[i]가 1이면 one에 1을 추가하고, 0이면 제거한 0의 개수를 나타내는 zero를 1만큼 증가시켰다.

이후에는 굳이 0을 one에 추가하고 제거하는 과정을 반복할 필요가 없다. 따라서 이진변환을 할 때 1인 경우에만 one에 추가하고 아닌 경우는 zero만 증가시킨다.

 

링크

programmers.co.kr/learn/courses/30/lessons/70129

코딩테스트 연습 - 이진 변환 반복하기

github.com/inbdni/Programmers/blob/master/level02/%EC%9D%B4%EC%A7%84%EB%B3%80%ED%99%98%EB%B0%98%EB%B3%B5%ED%95%98%EA%B8%B0.cpp

inbdni/Programmers