Nearest Beautiful Number (easy version)

题意
给你两个数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
#include
#include
#include
#include
#include
#define ll long long 
using namespace std;
const int maxn = 1e6 + 7;
set 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));
	}
}

你可能感兴趣的:(算法)