PAT Basic Level 1094 谷歌的招聘 （20 分）

题目链接:

https://pintia.cn/problem-sets/994805260223102976/problems/1071785997033074688

我的17分代码:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstdlib>

using namespace std;

bool isPrime(int n){
    float n_sqrt;
    if(n==2||n==3)  return true;
    if(n%6!=1&&n%6!=5)  return false;
    n_sqrt=floor(sqrt((float)n));
    for(int i=5;i<=n_sqrt;i+=6){
        if(n%i==0 | n%(i+2)==0)
            return false;
    }
    return true;
}


int main()
{
    int L,K;
    int flag=0;
    scanf("%d%d",&L,&K);
    string str;
    cin>>str;
    for(int i=0;i<str.size()-K-1;i++){//str.size()-K,测试点5出错
        string str_=str.substr(i,K);
        int a=atoi(str_.c_str());
        if(isPrime(a)){
            cout<<str_;
            flag=1;
            break;
        }
    }
    if(!flag)
        printf("404");
    return 0;
}