PAT乙级1094 谷歌的招聘

注意：前导零需要输出（测试点2）

#include<iostream>
#include<string>
#include<cmath>
#include<iomanip>
using namespace std;
bool isPrime(int a){
    if(a == 2 || a == 3 || a == 5 || a == 7)return true;
    if(a == 1 || a % 2 == 0)return false;
    for(int i = 3; i <= (int)sqrt(a); i += 2)if(a % i == 0)return false;
    return true;
}
int main(){
    int n,k,i,a;
    string s,str;
    cin>>n>>k;
    getchar();
    getline(cin,s);
    for(i = 0;i < n - k + 1; ++ i){
        str = s.substr(i,k);
        a = stoi(str.c_str());
        if(isPrime(a) == true){
            cout<<setw(k)<<setfill('0')<<a;
            return 0;
        }
    }
    cout<<"404";
    return 0;
}