F2. Nearest Beautiful Number (hard version) (贪心)

Posted 2021-09-09 Harris-H

F2. Nearest Beautiful Number (hard version) (贪心)

注意到一个结论：答案和$n$同位数，因为当所有位都为$9$必定满足。

然后就可以贪心，找到满足不同位数$\le k$的最大前缀$+1$的位置，显然此位需要加1，后面的位全部修改为$0$，然后修改$n$，直到$n$满足条件为止。

$ep:n=4561,k=3$

最大前缀是$456$，需要对$1$加不断$1$，直到$n=4564$满足情况返回。
$ep:n=4512,k=2$。

最大前缀$45$，第三位的$1$加$1$，后面置为$0$，$n=4520$，然后再重新模拟。

$n\to 4530\to 4540\to 4544$。

设$n$的位数是$m$位。

每位最多修改$10$次，每次遍历最多$m$位，最多遍历$m$次。

时间复杂度：$O(10m^2)$

#include <bits/stdc++.h>
using namespace std;
string solve(){
	string n;
	int k;
	cin >> n >> k;
	while (true)
	{
		set<char> s;
		for (auto c : n) s.insert(c);
		if (s.size() <= k) return n;
		s.clear();
		int ptr = 0;
		for (; ; ptr++)
		{
			s.insert(n[ptr]);
			if (s.size() > k)
			{
				while (n[ptr] == '9')
					ptr--;
				n[ptr]++;
				for (int i = ptr + 1; i < n.size(); i++)
					n[i] = '0';
				break;
			}
		}
	} 
}
int main(){
	int t;
	cin >> t;
	while (t--)
		cout << solve() << '\\n';	
	return 0;
}