20310 - 타노스

https://www.acmicpc.net/problem/20310

20310번: 타노스

생각해 보기

• 0과 1의 개수를 처음에 센다.
• 1이 앞에 오면 사전 순서로 뒤로 밀리게 된다.
• 0의 사전 순서가 1보다 앞선다.
• 0은 뒤에서 1은 앞에서 없앤다.

코드

#include<iostream>

using namespace std;

string s;
int zero=0,one=0;
bool arr[501];
void input(){
    cin>>s;
    fill(&arr[0],&arr[500],true);
    for(int i=0,len=s.length();i<len;++i){
        if(s[i]=='0')
            ++zero;
        else
            ++one;
    }
}

void solution(){
    for(int i=0,cnt=0;cnt<one/2;++i){
        if(s[i]=='1'){
            ++cnt;
            arr[i]=false;
        }
    }

    for(int i=s.length()-1,cnt=0;cnt<zero/2;--i){
        if(s[i]=='0'){
            ++cnt;
            arr[i]=false;
        }
    }

    for(int i=0,len=s.length();i<len;++i){
        if(arr[i])
            cout<<s[i];
    }
}

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    input();
    solution();

    return 0;
}