PAT 甲级1152 Google Recruitment (20 分)-蒲公英云

PAT 甲级1152 Google Recruitment (20 分)

输入格式

每个输入文件包含一个测试用例。
每种情况首先在一行中给出两个正整数：L（≤1,000）和K（<10），它们分别是给定数字的位数和要找到的质数。然后在下一行给出L位数字N。

输出格式

对于每个测试用例，在一行中连续打印N个连续数字中的第一个K位素数。如果不存在这样的数字，则输出404。
注意：前导零也必须算作K位的一部分。例如，要在200236中找到4位数的质数，可以使用0023。但是，由于前导零不在原始数字中，因此不能将第一个数字2视为解决方案0002。
在这里插入图片描述
题目就是从一个不超过一千位的数里面从前往后取连续的k位数，看该数是否是素数，如果是输出该数，不是的话输出404
有一个小测试蛮坑的，就是输出的时候如果连续的K位前面是0，也要一起输出。
举个例子，题目要求3位素数，结果是005，不能输出5，而是005

#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
bool isprime(int num)
{ 
    if(num==0||num==1)return false;
    else
    { 
        for(int i=2;i<=sqrt(num);i++)
        { 
            if(num%i==0)return false;
        }
        return true;
    }
}
int main()
{ 
    string s,ss;
    int a,b,temp;
    cin>>a>>b;
    cin>>s;//getline(cin,s);
    for(int i=0;i<=a-b;i++)
    { 
        ss=s.substr(i,b);
        temp=stoi(ss);
        if(isprime(temp))
        { 
            cout<<ss<<endl;
            return 0;
        }
    }
    cout<<"404"<<endl;
    return 0;
}