Codeforces 1102D Balanced Ternary String 思维

题意：

给出0，1，2序列，序列中的0，1，2数量不一样，但是0，1，2都可以被修改成其他数字，求将序列修改成0，1，2数量相等相等序列的最小修改次数，并输出修改后最小字典序的序列。

思路：

预处理序列的0，1，2的数字个数和位置，设0的数量是aa，1的数量是bb，2的数量是cc，此时分情况讨论：若aa==bb&&bb==cc直接输入，如果不是，使用三个双端队列，保存0，1，2元素的位置。先对2多的情况分析，将尽可能前面的2换成0，然后改成0的2的位置从保存2位置的队列弹出，进入保存0的位置的队列，从前弹出就从前加入，之后改成1的位置一样处理。如果此时1多于平均值，则将位置尽可能靠后的1换成2，将尽可能靠前的1换成0，并按照类似方法处理队列。最后按照相同的办法处理0。

AC代码：

#include <iostream>
#include <string>
#include <algorithm>
#include <deque>
using namespace std;
deque<int>num[3];
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int n;
	cin >> n;
	string a;
	cin >> a;
	for (int i = 0; i < n; ++i)
		num[a[i] - '0'].push_back(i);
	int k = n / 3;
	if (num[0].size() == num[1].size() && num[1].size() == num[2].size())
		cout << a << endl;
	else
	{
		if (num[2].size() > k)
		{
			while (num[2].size() > k&&num[0].size() < k)
			{
				int pos = num[2].front();
				num[2].pop_front();
				a[pos] = '0';
				num[0].push_front(pos);
			}
			while (num[2].size() > k&&num[1].size() < k)
			{
				int pos = num[2].front();
				num[2].pop_front();
				a[pos] = '1';
				num[1].push_front(pos);
			}
		}
		if (num[1].size() > k)
		{
			while (num[1].size() > k&&num[2].size() < k)
			{
				int pos = num[1].back();
				num[1].pop_back();
				a[pos] = '2';
				num[2].push_back(pos);
			}
			while (num[1].size() > k&&num[0].size() < k)
			{
				int pos = num[1].front();
				num[1].pop_front();
				a[pos] = '0';
				num[0].push_front(pos);
			}
		}
		if (num[0].size() > k)
		{
			while (num[0].size() > k&&num[2].size() < k)
			{
				int pos = num[0].back();
				num[0].pop_back();
				a[pos] = '2';
				num[2].push_back(pos);
			}
			while (num[0].size() > k&&num[1].size() < k)
			{
				int pos = num[0].back();
				num[0].pop_back();
				a[pos] = '1';
				num[1].push_back(pos);
			}
		}
		cout << a << endl;
	}
	//system("pause");
	return 0;
}