题意:给出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;
}