PTA 乙级 1104 天长地久 C语言 高效代码

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

代码长度限制

16 KB

时间限制

3000 ms

内存限制

64 MB

可以证明，末尾一定是连续的9（且至少两个），这个其他有大佬通过数学方法已经证明了，所以可以通过模拟末尾的9来优化代码

#include<stdio.h>
#include<math.h>
#include<stdlib.h>
typedef struct{ int n, a; }node;
int cmp(const void *a, const void *b){
    node *x = (node*) a, *y = (node*) b;
    return (x -> n != y -> n) ? x -> n - y -> n : x -> a - y -> a;
}
int gcd(int a, int b){
    return b ? gcd(b, a % b) : a;
}
int check(int num){
    if (num <= 2) return 0;
    for (int i = 2; i < num; ++i)
        if (num % i == 0) return 0;
    return 1;
}
int sum(int n){
    int number = 0;
    while (n) {
        number += n % 10;
        n /= 10;
    }
    return number;
}
int main() {
    int n, k, m, cnt, num9, nums;
    node record[(int) 1e5];
    scanf("%d", &n);
    for (int i = 1 ; i <= n; i++) {
        cnt = 0;
        printf("Case %d\n", i);
        scanf("%d %d", &k, &m);
        if(m >= k * 9) {
            printf("No Solution\n");
            continue;
        }
        for (num9 = 2; m - num9 * 9 + 1 > 0; ++num9) {  //可以证明，末尾至少有两个9并且末尾的9是连续的，思路是构造末尾的连续9
            if (!check(gcd(m, m - num9 * 9 + 1))) continue;
            int a = (int) (pow(10, k - 1) + pow(10, num9)) - 1;     //初始数字 = (1ek + 构造的9)
            int end = (int) pow(10, k), step = (int) pow(10, num9); //结束数字 = 1e(k + 1), 步长 = 1
            for (nums = 0; a < end; a += step, ++nums) {
                if (nums % 10 == 9 || sum(a) != m) continue;    //模拟的关键，如果模拟的次数是9的整数倍，则a + 1不会满足
                record[cnt].n = sum(a + 1);
                record[cnt++].a = a;
            }
        }
        qsort(record, cnt, sizeof(node), cmp);
        for (int x = 0; x < cnt; x++)
            printf("%d %d\n", record[x].n, record[x].a);
        if (!cnt)
            printf("No Solution\n");
    }
    return 0;
}

 运行截图