PAT A 2019.9 7-1 Forever (20分)

一、题目概述

存在整数A， 满足以下条件:
1、A共K位
2、A各位数字和为m
3、A + 1各位数字和为n
4、m和n的最大公因数为大于2的素数

要求：依据给定K和m，输出所有满足条件的n和A。输出按照n和A升序

二、思路

设A各位数字之和为m，则A + 1各位数字之和n有以下可能：
1、A末尾数字不是9，显然：n = m + 1
2、A尾部有k位连续的9：n = m + 1 - k * 9
归纳后，n可能的取值为：k >= 0, n = m + 1 - k * 9

因此，遍历所有可能的n，当m，n的最大公因数为大于2的素数时，则n取值合法，且可以确定，A尾部有k个9。
进行深度优先搜索，找到所有前K - k位和为m - k * 9（因为尾部有k个连续的9），且首位不为0（开头为零则数字小于K位），末尾不为9的序列（尾部为9则A有个看k + 1个连续的尾数9，和不为n），拼接上k位9，即为所有合法的A。

三、代码（PAT暂未收录，不保证正确）

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> A;
bool flag;
int gcd(int a, int b)
{
	return !b ? a : gcd(b, a % b);
}
bool isprime(int x)
{
	if(x < 3)
		return false;
	for( int i = 2; i * i <= x; ++i )
		if( x % i == 0 )
			return false;
	return true;
}
void dfs(int rem, int pre_digit, int post_digit, int n)
{
	if( A.size() == pre_digit )
	{
		if( !rem )
		{
			flag = true;
			printf("%d ", n);
			for( int i = 0; i < A.size(); ++i )
				printf("%d", A[i]);
			for( int i = 0; i < post_digit; ++i )
				printf("9%s", i == post_digit - 1 ? "\n":"");
		}
		return;
	}
	for( int i = (A.size() ? 0:1); i <= (A.size() == pre_digit - 1 ? 8:9); ++i )
		if( rem >= i )
		{
			A.push_back(i);
			dfs(rem - i, pre_digit, post_digit, n);
			A.pop_back();
		}
}
int main()
{
	int N;
	scanf("%d", &N);
	for( int i = 1, K, m; i <= N; ++i )
	{
		scanf("%d %d", &K, &m);
		flag = false;
		printf("Case %d\n", i);
		for( int k = (m + 1) / 9; k >= 0; --k )
			if( isprime(gcd(m, m + 1 - k * 9)) )
				dfs(m - k * 9, K - k, k, m + 1 - k * 9);
		if( !flag )
			printf("No Solution\n");
	}
}