[프로그래머스/level2/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>
#include <algorithm>
using namespace std;

vector<int> solution(string s) {
    vector<int> answer(2);
    while(s!="1"){
        int len=s.size();
        s.erase(remove(s.begin(), s.end(), '0'), s.end());
        answer[1]+=len-s.size();
        len=s.size();
        s="";
        while(len>0){
            s=to_string(len%2)+s;
            len/=2;
        }
        answer[0]++;
    }
    return answer;
}

 

다시 짚고 넘어가기 :

s.erase(remove(s.begin(),s.end(),'0'),s.end()); 부분에서 

remove는 일치하는 모든 요소를 삭제하지만 길이는 그대로이고(iterator는 해당요소를 삭제하여 뒤로보낸 위치를 가리킴)

erase는 해당하는 인덱스 범위를 삭제하여 길이도 같이 줄어든다.

 

도움이 된 글 :

egloos.zum.com/h2ostudio/v/4343561

[STL] erase 와 remove