PAT Basic 1094 谷歌的招聘 (20 分)

20 5
23654987725541023819

输出样例 1：

49877

输入样例 2：

10 3
2468024680

输出样例 2：

404

#include <iostream>
#include <sstream>
#include <cmath>
using namespace std;
bool isPrime(int a){
    if(a==2||a==3) return true;
    for(int i=2;i<=sqrt(a);i++)
        if(a%i==0) return false;
    return true;
}
int main()
{
    int len,max_len;
    cin>>len>>max_len;
    int capacity=1;
    for(int i=0;i<max_len;i++){
        capacity*=10;
    }
    string s;int tmp;string ans="";
    cin>>s;
    for(int i=0;i<(s.length()-max_len+1);i++){
        stringstream ss;
        ss<<s.substr(i,max_len);
        ss>>tmp;
        if(isPrime(tmp)) {
            ans=s.substr(i,max_len);
            break;
        }
    }
    if(ans!="") {
        cout<<ans;
    }
    else cout<<404;
    system("pause");
    return 0;
}