PTA谷歌的招聘

2004 年 7 月，谷歌在硅谷的 101 号公路边竖立了一块巨大的广告牌（如下图）用于招聘。内容超级简单，就是一个以 .com 结尾的网址，而前面的网址是一个 10 位素数，这个素数是自然常数 e 中最早出现的 10 位连续数字。能找出这个素数的人，就可以通过访问谷歌的这个网站进入招聘流程的下一步。

自然常数 e 是一个著名的超越数，前面若干位写出来是这样的：e = 2.718281828459045235360287471352662497757247093699959574966967627724076630353547594571382178525166427427466391932003059921... 其中粗体标出的 10 位数就是答案。

本题要求你编程解决一个更通用的问题：从任一给定的长度为 L 的数字中，找出最早出现的 K 位连续数字所组成的素数。

输入格式：

输入在第一行给出 2 个正整数，分别是 L（不超过 1000 的正整数，为数字长度）和 K（小于 10 的正整数）。接下来一行给出一个长度为 L 的正整数 N。

输出格式：

在一行中输出 N 中最早出现的 K 位连续数字所组成的素数。如果这样的素数不存在，则输出 404。注意，原始数字中的前导零也计算在位数之内。例如在 200236 中找 4 位素数，0023 算是解；但第一位 2 不能被当成 0002 输出，因为在原始数字中不存在这个 2 的前导零。

输入样例 1：

20 5
23654987725541023819

输出样例 1：

49877

输入样例 2：

10 3
2468001680

输出样例 2：

404

 以下是我的答案

#include<stdio.h>
#include<math.h>
#define MAX 1000
long prime(long n);//判断是否为素数，效率较高
int main()
{
    int A[MAX] = { 0 };
    long L, K, sum = 0;
    scanf("%ld %ld", &L, &K);
    getchar();
    for (int i = 0; i < L; i++)
    {
        char a;
        scanf("%c", &a);
        A[i] = (int)(a - '0');
    }
    //读入数组
    for (int i = 0; i < MAX - K; i++)
    {
        for (int j = K - 1; j >= 0; j--)
            sum = A[i + K - 1 - j] * pow(10, j) + sum;
        if (prime(sum) == 1)
        {
            int z=0;
            //以下代码实现动态的输出，如K=4时，输出0023，而不是23
            for(int temp=sum;z<K;z++)
            {
                if(temp==0)
                    break;
                temp/=10;
            }
            for(int i=K-z;i>0;i--)
                printf("0");
            printf("%d", sum);
            return 0;
        }
        //        printf("%d\n",sum);
        sum = 0;
    }
    //遍历数组，每次检索K位数，如果为素数，输出
    printf("404");
}

long prime(long n) {//是否为素数
    if (n < 2)
        return 0;
    for (long i = 2; i * i <= n; i++)
        if (n % i == 0) return 0;
    return 1;
}