题意
给你两个数n,k,让你找到大于等于n的最小的k-beautiful数。
k-beautiful数就是数字的种类数小于等于k的数。
n的范围是1<=n<=1e9.
k的范围是1<=k<=2.
举个例子 比如 177890 2,得到的最小的2-beautiful是181111.
思路
刚开始的时候k=1的时候很简单,就直接模拟就可以了,但是当k=2的时候,刚开始我也想直接模拟,可是发现总会少考虑情况,最后看了题解发现我们可以把所有的1-beautiful数和2-beautiful都直接预处理出来,存起来。
代码
#include<cstdio>
#include<iostream>
#include<cstring>
#include<map>
#include<set>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 1e6 + 7;
set<ll > s1, s2;
void pre() {
for (ll i = 0; i <= 9; ++i) {
ll num = 0;
for (ll j = 1; j <= 10; ++j) {
num = num * 10 + i;
s1.insert(num);
s2.insert(num);
}
}
for (ll i = 0; i <= 9; ++i){
for (ll j = i + 1; j <= 9; ++j){
for (int k = 0; k < (1 << 10); ++k){
ll num = 0;
for (int l = 0; l <= 9; ++l){
if ((k >> l) & 1)num = num * 10 + i;
else num = num * 10 + j;
s2.insert(num);
}
}
}
}
}
int main() {
pre();
int T;
scanf("%d", &T);
while (T--) {
ll n;
int k;
scanf("%lld %d", &n, &k);
if (k == 1)printf("%lld\n", *s1.lower_bound(n));
else printf("%lld\n",*s2.lower_bound(n));
}
}