Codeforces Round #739 (Div. 3) F2. Nearest Beautiful Number (hard version)

Codeforces Round #739 (Div. 3)的其他题解在这

F2. Nearest Beautiful Number (hard version)

---

题目大意：

一个数各数位出现的数字个数小于k的数被称为美丽的数
给出一个数n
求大于等于n的最小美丽的数
1<=k<=10

思路：
由于k范围扩大了，我们就不能用F1的预处理的方法来做了
这里我们可以直接dfs
有点类似于数位DP
可以看下面代码里的解释

AC代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<map>
#include<set>
#include<algorithm>
#define ll long long 
using namespace std;
const int maxn = 1e6 + 7;
int ans, k, n;
string x;
int calc(int x) {//计算x的二进制有多少个1
	int tot = 0;
	while (x) {
		if (x % 2)tot++;
		x >>= 1;
	}
	return tot;
}//f用来记录当前已经选过的数，10位二进制，1表示选了，0表示没选，eg：111表示选了0，1,2
void dfs(int t,int limit,int sum,int f) {//t表示当前位数，sum表示当前的数
	int len = x.length();//limit表示前面选的数是否都是等于原数
	if (ans)return;//如果等于原数，那么后面不能小于原数，如果前面不等于，则后面随意
	if (t == len) {
		ans = sum;
		return;
	}//循环起点与limit有关
	for (int i = limit ? x[t] - '0' : 0; i <= 9; ++i) {
		if (calc(f | (1 << i)) > k)continue;//如果选的数超过k
		if (limit && i == x[t] - '0')dfs(t + 1, 1, sum * 10 + i,f|(1<<i));
		else dfs(t + 1, 0, sum * 10 + i,f|(1<<i));
	}
}
int main() {
	int T;
	scanf("%d", &T);
	while (T--) {
		cin >> x >> k;
		ans = 0;
		dfs(0,1,0,0);
		printf("%d\n",ans);
	}
}