Codeforces Round #705 (Div. 2)-C. K-beautiful Strings

C. K-beautiful Strings

贪心

---

• 题意
  输入n，k，代表待处理字符串的长度和美丽数字，再输入只由小写字母构成的字符串s。定义：如果一个字符串中的任意一个字母的数量都是k的整数倍，则称这个字符串是美丽的。
  现可对待处理的字符串进行任意次数的元素替换操作。
  求：让原串变成美丽字符串，且字典序大于等于原串的前提下，在所有解集中字典序最小的答案。

---

• 思路
  毫无疑问，首先对原串的每个字母数量计数。
  预处理，计算出在可以添加字母的条件下，把所有字母数量变成k的整数倍，需要的添加的字母总数量sum。
  倒序遍历+贪心：rep（pos，n，1）
  【1，pos-1】位置上的字母不动，把【pos，n】的字母划分为待转换状态（按需要替换），首次符合条件的就是字典序最小的答案。怎么按需要来替换？怎么判断是否符合？
  1、变量sum的值根据遍历更新，它恒表示只添加字母（并非修改）的前提下，将字符串【1，pos-1】中字母数补成k整数倍，需要在后面额外添加的字母总数。
  2、优先对当前位置pos的字符处理，枚举替换成s【pos】的下一个字母，即s【pos】-‘a’+1至25。
  3、第二步将s【pos】替换后，注意对应字母数量和sum需要同步更新，接着对【pos+1，n】这段上进行填充（等效于替换），
  注：此时字符串分成了三段，【1，pos-1】是不改变的，【pos，pos】是当前位贪心枚举，【pos+1，n】是待修改的部分（可看作为空位），它提供位置给第一段，即填充sum个字母后，与【1，pos-1】中的字母数量相加，数量都符合k的整数倍的条件。
  【pos-1，n】这段待填充串，不仅需要sum个位置用来填补第一段，多余位置要填满字母a。因为需要字典序最小，字母a全部放在sum个填充字母前。证明：因为其他字母数量已经是符合要求的，则填充的字母a数量也必是k的整数倍。
  4、整个字符有解的前提肯定是n%k==0，且第三步中可以填充的前提当然是pos+sum<=n，另外注意第二步的枚举中需要对操作回溯。

---

• 代码

ll n,k,cnt[30];
char s[maxn];
ll get(int a){return (k-a%k)%k;}
void solve()
{
	cin>>n>>k;
	cin>>s+1;
	if(n%k!=0)	Cneg
	
	mem(cnt,0);
	for(int i=1;i<=n;i++)
		cnt[s[i]-'a']++;
	ll sum=0;
	for(int i=0;i<26;i++)
		sum += get(cnt[i]);
	
	if(sum%k!=0)	cout<<-1<<endl;
	else if(sum==0)	cout<<s+1<<endl;
	else
	{
		for(int i=n;i>=1;i--)
		{
			sum -= get(cnt[s[i]-'a']);
			cnt[s[i]-'a']--;
			sum += get(cnt[s[i]-'a']);
			
			for(int j=s[i]-'a'+1;j<26;j++)
			{
				ll need = sum;
				need -= get(cnt[j]);
				cnt[j]++;//
				need += get(cnt[j]);
				
				if(i+need <= n)
				{
					for(int pre=1;pre<=i-1;pre++)	cout<<s[pre];
					cout<<char(j+'a');
					for(int bak=i+need+1;bak<=n;bak++)	cout<<'a';
					for(int c=0;c<26;c++)
					{
						ll num = get(cnt[c]);
						while(num--) cout<<char(c+'a');
					}
					cout<<endl;
					return ;
				}
				cnt[j]--;//
			}
		}
	}
}