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

이진 변환 반복하기 / Lv.2

---

문제  설명 )

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>
#include <bitset>
 
using namespace std;
 
vector<int> solution(string s) {
        
    int count1 = 0;
    int count2 = 0;
    
    // 처음 01110 과 같은 형태가 주어질 수 있으므로, 
    for(auto &ele: s)
    {
        if (ele == '0') count1++;
        else break;
    }
    
    while(1)
    {
        // 1이 되었다면 반복문 끝
        if (s == "00000000000000000001") break;
 
        string temp = "";
        bool flag = 0;
        for (int i = 0; i < s.length(); i++)
        {
            if (flag && s[i] == '0') count1++;
            else if (s[i] == '1')
            {
                flag = 1;
                temp += '1';
            }
        }
 
        // bitset의 최대 길이 20 
        s = bitset<20>(temp.length()).to_string();
        count2++; 
    }
    
    return vector<int> {count2, count1};
}

---

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/70129

프로그래머스