1. 程式人生 > >Balanced Ternary String

Balanced Ternary String

https://codeforces.com/contest/1102/problem/D

題解:

/*
*@Author:   STZG
*@Language: C++
*/
#include <bits/stdc++.h>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<bitset>
#include<queue>
#include<deque>
#include<stack>
#include<cmath>
#include<list>
#include<map>
#include<set>
//#define DEBUG
#define RI register int
using namespace std;
typedef long long ll;
//typedef __int128 lll;
const int N=300000+10;
const int MOD=1e9+7;
const double PI = acos(-1.0);
const double EXP = 1E-8;
const int INF = 0x3f3f3f3f;
int t,n,m,k,q;
int ans,cnt,flag,temp;
int a[3];
char str[N];
int main()
{
#ifdef DEBUG
	freopen("input.in", "r", stdin);
	//freopen("output.out", "w", stdout);
#endif
    scanf("%d",&n);
    scanf("%s",str);
    //while(t--){}
    for(int i=0;i<n;i++){
        if(str[i]=='0')
            a[0]++;
        else if(str[i]=='1')
            a[1]++;
        else if(str[i]=='2')
            a[2]++;
    }
    m=n/3;
    if(a[0]<m){
        if(a[2]>m)
        for(int i=0;i<n;i++){
            if(str[i]=='2'){
                str[i]='0';
                a[0]++;
                a[2]--;
                if(a[0]==m||a[2]==m)
                    break;

            }
        }
        if(a[1]>m&&a[0]<m)
        for(int i=0;i<n;i++){
            if(str[i]=='1'){
                str[i]='0';
                a[0]++;
                a[1]--;
                if(a[0]==m||a[1]==m)
                    break;

            }
        }
    }
    if(a[2]<m){
        if(a[1]>m)
        for(int i=n-1;i>=0;i--){
            if(str[i]=='1'){
                str[i]='2';
                a[2]++;
                a[1]--;
                if(a[2]==m||a[1]==m)
                    break;

            }
        }
        if(a[0]>m&&a[2]<m)
        for(int i=n-1;i>=0;i--){
            if(str[i]=='0'){
                str[i]='2';
                a[2]++;
                a[0]--;
                if(a[2]==m||a[0]==m)
                    break;

            }
        }
    }
    if(a[1]<m){
        if(a[0]>m)
        for(int i=n-1;i>=0;i--){
            if(str[i]=='0'){
                str[i]='1';
                a[1]++;
                a[0]--;
                if(a[0]==m||a[1]==m)
                    break;

            }
        }
        if(a[2]>m&&a[1]<m)
        for(int i=0;i<n;i++){
            if(str[i]=='2'){
                str[i]='1';
                a[1]++;
                a[2]--;
                if(a[1]==m||a[2]==m)
                    break;

            }
        }
    }
    cout<<str<<endl;
    //cout << "Hello world!" << endl;
    return 0;
}