1104. 天长地久 (20 分)

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

我以前也是纯C战士，但是后来做题时，发现有些题目纯C实现起来好麻烦，与此同时发现C++是真的好用，它是真的好用。

这个题目用深度搜索+剪枝可以解决。可以先写个demo，循环输出000到999，然后加上剪枝和判断条件，题目就完成了。

demo如下（非答案）：

把demo搞明白了，后面自己也写得出来了。

#include<cstdio>
#include<set>
#include<map>
#include<cmath> 
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

int num[10];
//cnt 当前第cnt位  sum 前cnt位的数字累计之和 K查找到多少位 
void dfs(int cnt,int sum,int K){
	//边界条件 到多少层跳出循环，不然无限向下，死循环了 
	//输出 第3位赋值之后 已经达到目标 3位数 
	//第三层 在3位数字赋值之后 还会执行cnt+1，会再进入下一层，而此时作为第四层的开头 cnt为 4 所以为 cnt(4) == K(3)+1 为结束条件 
	if(cnt== (K+1)){
		for(int i=1;i<=K;i++){
			cout<<num[i];
		} 
		cout<<" sum = "<<sum<<endl;
		return;
	}
	
	for(int i=0;i<=9;i++)
	{
		//赋值第cnt位为 i 循环调用 就会把cnt位 分别复制位 0 - 9 挨个往下查询 
		num[cnt]=i;
		//cnt+1 代表搜索下一位数字 sum + i 为累计和  K仅传递值 不做改变，定义为全局变量也行 
		dfs(cnt+1,sum+i,K);
	}
}

int main(){
	//从第1位开始搜索，初始和位0，总共3位 
	dfs(1,0,3);
	return 0;
}

AC代码如下：​​​​​​

#include <cstdio>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
int num[15];
//标记 ture表示没有找到解
bool flag = true;
//有序set容器ans pair的作用是可以按照第一个int升序，若相等则按第二个int升序排序
set< pair<int, int>   > ans;

bool isPrime(int num)
{
	//题目要求大于2的素数
	if (num <= 2)
		return false;
	int n = sqrt(num);
	for (int i = 2; i <= n; i++)
		if (num % i == 0)
			return false;
	return true;
}
//求最大公约数
int gcd(int x, int y)
{
	int z = y;
	while (x % y != 0)
	{
		z = x % y;
		x = y;
		y = z;
	}
	return z;
}

int converse(int K)
{
	int sum = 0;
	//从 第 1 位 到 第 N 位 
	for (int i = 1; i <= K; i++)
	{
		sum = sum * 10 + num[i];
	}
	return sum;
}

int getSum(int N)
{
	int sum = 0;
	while (N > 0)
	{
		sum += N % 10;
		N = N / 10;
	}
	return sum;
}



void dfs(int cnt, int K, int sum, int m)
{

	//如果检索到了最后一位
	if (cnt > K)
	{
		//将数组转为整数 A 
		int A = converse(K);
		//求A+1的各位数之和 n
		int n = getSum(A + 1);
		//如果A的各位数实际的和sum ==  目标和 m， 且 m 和 n 的最小公约数是大于2的质数
		if (sum == m && isPrime(gcd(m, n)))
		{
			ans.insert({n, A}); //将符合条件的n和A插入到ans中
			//找到解了
			flag = false;
		}
		return;
	}
    //【 剪枝关键 】
	//如果现有前cnt位数字之和 加上后面尚未列举的数字全部置为 9 的情况下
	//仍然无法使得这个数字的所有位数之和大于等于 m 就无需继续往下查找了 
	//注意这里是 K + 1 -cnt 而不是 K - cnt ，因为当第一位赋为 1 后，进入第二层 dfs ，此时cnt = 2 但是 实际上 第二位还尚未赋值
	//真正赋值是在后面的for循环进行的 所以此时实际上只有第一位 赋值了 累计和 sum 中也只有第一位的和  所以 后续还有 6(K) - 2(此时cnt) + 1 个数字 可以计算为9 
	if (sum + (K + 1 - cnt) * 9 < m)
	{
		return;
	}

	for (int i = 0; i <= 9; i++)
	{
		num[cnt] = i;
		//首位不为 0 
		if(cnt ==1 && i==0){
			continue;
		}
		dfs(cnt + 1, K, sum + i , m);
	}
}

int main()
{
	int N, K, m;
	cin >> N;
	for (int t = 1; t <= N; t++)
	{
		//清空上次计算记录 
		ans.clear();
		//重置  标记 ture表示没有找到解
		flag = true;
		cin >> K >> m;
		cout << "Case " << t << endl;
		//从第一位开始查找，总共K位，初始和位0，目标是和要等于m 
		dfs(1, K, 0, m);
		if (flag)
		{
			cout << "No Solution" << endl;
		}
		else
		{
			//cout<<ans; 注意  <int, int> > 这里 > 和 > 不能相邻 否则会识别为 >>  
			for (set< pair<int, int> >::iterator it = ans.begin(); it != ans.end(); it++) //用迭代器遍历整个容器
			{
				cout << (*it).first << " " << (*it).second << endl;
			}
		}
	}
	return 0;
}