codeforces round 531(div3) D. Balanced Ternary String

维护两个数组，一个num数组记录还有多少个i字符，一个b数组记录有多少个i字符需要替换。用ans数组记录需要替换的字符集合（从小到大）。在b[i[>0的情况下，只有两种情况s可以替换，一种情况是ans数组的首位字典序比s当前的字符小，替换使得字典序变小

还有一种是num[i]等于b[i]，所有i字符必须全部被替换

#include<bits/stdc++.h>
using namespace std;
char s[300010];
int num[3];
int b[3];
int ans[300010];
int tot;
int main()
{
	int n;
	scanf("%d",&n);
	scanf("%s",s);
	for(int i=0;s[i];i++)
	{
		num[s[i]-'0']++;//1,2,3的数量 
	}
	for(int i=0;i<=2;i++)
	{
		b[i]=num[i]-n/3;//需要减少的数量 
	}
	for(int i=0;i<=2;i++)
	{
		if(b[i]<0)
		{
			while(b[i]<0)
			{
				ans[++tot]=i;
				b[i]++;
			}
		}
	}
	int now=0;
	for(int i=0;s[i];i++)
	{
		int index=s[i]-'0';
		if(b[index]>0&&( index>ans[now+1]||b[index]==num[index] ) )
		{
			s[i]=ans[++now]+'0';
			b[index]--;
		}
		num[index]--;
		if(now==tot)
		  break;
	}
	printf("%s\n",s);
}

　　

转载于:https://www.cnblogs.com/lishengkangshidatiancai/p/10252497.html