“天长地久数”是指一个 K 位正整数 A,其满足条件为:A 的各位数字之和为 m,A+1 的各位数字之和为 n,且 m 与 n 的最大公约数是一个大于 2 的素数。本题就请你找出这些天长地久数。
输入格式:
输入在第一行给出正整数 N(≤5),随后 N 行,每行给出一对 K(3<K<10)和 m(1<m<90),其含义如题面所述。
输出格式:
对每一对输入的 K 和 m,首先在一行中输出 Case X
,其中 X
是输出的编号(从 1 开始);然后一行输出对应的 n 和 A,数字间以空格分隔。如果解不唯一,则每组解占一行,按 n 的递增序输出;若仍不唯一,则按 A 的递增序输出。若解不存在,则在一行中输出 No Solution
。
输入样例:
2
6 45
7 80
输出样例:
Case 1
10 189999
10 279999
10 369999
10 459999
10 549999
10 639999
10 729999
10 819999
10 909999
Case 2
No Solution
#include <iostream>
#include <cmath>
#include <unordered_set>
#include <set>
using namespace std;
bool tag;//标记是否存在答案
unordered_set<int> st;//存储每一个m对应的所有可能满足条件的质数
using LL = long long;
int gcd(int a, int b) //辗转相除法求最大公约数最简模版
{
return b == 0 ? a : gcd(b, a % b);
}
bool isPrime(int x) {
if (x <= 1) return false;
int sq = sqrt(x);
for (int i = 2; i <= sq; i++) {
if (x % i == 0) return false;
}
return true;
}
set<pair<int, LL>> res;
void dfs(int u, int K, int m, int sum, LL A)//当前位数,目标位数,目标位数和,当前位数和 ,当前数
{
if (u > K) return;
if (sum + (K - u) > m) return;
if (sum + (K - u) * 9 < m) return;
/*
* 这里不写的话会在最后一个测试点超时
* 这里是剪枝操作,就是若剩下的数全取9也取不到目标位数和m式就可以return,或者当前已经大于m
*/
if (u == K) {
if (sum == m) {
LL val = A + 1;
int n = 0;
do {
n += val % 10;
val /= 10;
} while (val);
if (st.count(n)) //count是查找单个元素return 1 or 0,用find也可以
{
res.insert({ n, A });
tag = true;
}
}
return;
}
for (int i = 0; i <= 9; i++) {
if (!u && !i) continue;//枚举的最高位数不能是0
dfs(u + 1, K, m, sum + i, A * 10 + i);
}
}
int main() {
int N;
cin >> N;
for (int i = 1; i <= N; i++) {
printf("Case %d\n", i);
tag = false;
int K, m;
cin >> K >> m;
st.clear();
res.clear();
//预处理
for (int i = 1; i <= 90; i++) {
auto t = gcd(i, m);//将所有可能满足条件的i放入集合搜索
if (t > 2 && isPrime(t)) {
st.insert(i);
}
}
dfs(0, K, m, 0, 0);
if (!tag) puts("No Solution");
else {
for (auto x : res) {
printf("%d %lld\n", x.first, x.second);
}
}
}
return 0;
}
参考博客