PAT (Basic Level) Practice 1094

1094 谷歌的招聘 （20 分）

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
2468024680

输出样例 2：

404

分析：题目数据规模太大，打表必定超时，直接用函数来判素数。遍历数字，每次取K位判断即可。输出时注意前导零也需要一起输出。

代码：

#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
#include<string>
using namespace std;
int isPrime(int n) {
	int i = 2;
	for (; i <= sqrt(n); i++) {
		if (n % i == 0) {
			break;
		}
	}
	if (i > sqrt(n)) {
		return 1;
	} else {
		return 0;
	}
}
int main() {
	int L, K;
	cin >> L >> K;
	string s;
	cin >> s;
	vector<int> v(s.size());
	for (int i = 0; i < L; i++) {
		v[i] = s[i] - '0';
	}					  
	int flag = 0;
	for (int i = 0; i <= L - K; i++) {
		int sum = 0;
		string st = "";
		for (int j = 0; j < K; j++) {
			sum = sum * 10 + v[i + j];
			st += v[i + j] + '0';
		}
		if (isPrime(sum) == 1) {
			flag = 1;
			cout << st << endl;
			break;
		}
	}
	if (flag == 0) {
		cout << 404 << endl;
	}
	return 0;
}