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

문제

---

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

풀이

---

주어진 string이 “1”이 될 때까지 이진 변환을 계속하고 이진 변환의 횟수와 제거된 0의 개수를 리턴해야한다.

따라서 0을 제거, 카운트하고 이를 이진 변환하는 것을 string이 “1”이 될 때까지 반복한다.

다만, 제한 사항으로 “s의 길이는 1 이상 150,000 이하입니다.” 라는 조건으로 인해 int 범위를 초과할 수 있으므로, bitset을 사용하여 변환한다.

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

using namespace std;

vector<int> solution(string s) 
{
    vector<int> answer;

    int loopCnt = 0;
    int eraseCnt = 0;
    int countOne = 0;
    bool isEnd = false;

    while (!isEnd)
    {
        if (loopCnt == 0)
        {
            countOne = count(s.begin(), s.end(), '1');
            eraseCnt += (s.size() - countOne);
        }

        else
        {
            int i = 0;
            for (; i < s.size(); i++)
            {
                if (s[i] == '1')
                {
                    break;
                }
            }

            countOne = count(s.begin() + i, s.end(), '1');
            eraseCnt += count(s.begin() + i, s.end(), '0');

        }

        bitset<32> bt = countOne;
        s = bt.to_string();
        loopCnt++;

        if (bt.count() == 1 && s[s.size()-1] == '1')
        {
            answer.push_back(loopCnt);
            answer.push_back(eraseCnt);
            break;
        }
    }

    return answer;
}