PTA刷题日记——1104天长地久（dfs）

“天长地久数”是指一个 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;
}

参考博客

1104 天长地久 (20 分)_1104天长地久_SYaoJun的博客-CSDN博客