【PAT乙级】1094 谷歌的招聘 (20分)

输入格式：
输入在第一行给出 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

思路基本没有问题，但是小问题不断，就是substr函数的用法我给记错了，substr（a,b），a是截取开始的下标没问题，我以为b是结束的下标，但不是，而是截取的字符长度。
还有就是素数判断没有考虑0和1，其次sqrt(n)可以用n<i*i代替，省去了头文件cmath了

代码

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int sushu(long n){
	if(n==0||n==1)return 0;
	for(int i=2;i<=sqrt(n);i++){
		if(n%i==0&&i<n) return 0;
	}
	return 1;
}
int main(){
	int l,k;
	cin>>l>>k;
	string s;
	cin>>s;
	int flag=0;
	for(int i=0;i<=l-k;i++){
		string s1=s.substr(i,k);
		int n=sushu(stol(s1));
		if(n==1){
			cout<<s1;
			flag=1;
			break;
		}
	}
	if(flag==0)cout<<"404";
}