Balanced Ternary String
阿新 • • 發佈:2019-01-12
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; }