PAT乙級 - 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：

輸入樣例 2：

10 3
2468024680
           

輸出樣例 2：

注意點：

這題思路很清晰，我主要談談測試點4。測試點4其實就是類似下面這種 K > L 的資料。

3 5
123
           

結果應輸出404，倘若循環頭中寫的是

for (int i = 0; i <= str.size() - k; i++)

，那麼會輸出23。問題在于str.size()是unsigned型，和int一起運算時，int都會被當成unsigned。以上面的資料為例，

str.size() - k

在64位系統中将等于264-2，是以會進入循環。而當string的substr方法參數中的偏移量超過string的長度時，隻截取至string末尾，是以最終輸出23。

解決方法很簡單，直接用L代替str.size()，或者強制轉換一下格式都是可以的，例如循環頭寫成

for (int i = 0; i <= (int)str.size() - k; i++)

#include <bits/stdc++.h>
using namespace std;
bool IsPrime(int num)
{
	if (num == 1)
		return false;
	for (int i = 2; i <= sqrt(num); i++)
		if (num % i == 0)
			return false;
	return true;
}
int main()
{
	ios::sync_with_stdio(false);
	int l, k;
	string str;
	cin >> l >> k;
	cin >> str;
	for (int i = 0; i <= l - k; i++)
	{
		string s = str.substr(i, k);
		if (IsPrime(atoi(s.c_str())))
		{
			cout << s;
			return 0;
		}
	}
	cout << "404";
	return 0;
}