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

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

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

이진 변환 반복하기

문제

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. s가 "1"이 될 때 까지 아래의 과정을 반복한다.

2. s의 모든 "0"을 제거한다.

3. 제거한 후의 s의 길이를 n이라고 하면 n을 2진수로 바꾼다.

4. 위의 과정을 반복한다.

 

#include <string>
#include <vector>

using namespace std;

vector<int> solution(string s) {
    vector<int> answer;
    size_t pos;
    int erase_count = 0;
    int trans_count = 0;
    int num;
    string transbin;
    
    while (s != "1") {
        transbin = "";
        while ((pos = s.find("0")) != string::npos) {
            s.erase(pos, 1);
            erase_count++;
        }
        num = s.length();
        
        while (num > 1) {
            transbin.insert(0, to_string(num % 2));
            num /= 2;
        }
        transbin.insert(0, to_string(num));
        s = transbin;
        trans_count++;
    }
    
    answer.push_back(trans_count);
    answer.push_back(erase_count);
    
    return answer;
}