C++求助

#include<iostream>
using namespace std;
bool is_prime(int a){
    if (a==2)return true;
    for (int i=2;i<a;i++){
        if (a%i==0)return false;
    }
    return true;
}

int main(){
    int a;
    int cnt=0,i=1;
    cin>>a;
    while (cnt!=a){
        i++;
        if (is_prime(i))cnt++;
    }
    cout<<i;
    return 0;
}

#include<iostream>
using namespace std;
bool is_prime(int a){
    if (a==2)return true;
    for (int i=2;i*i<a;i++){//差别在这里
        if (a%i==0)return false;
    }
    return true;
}

int main(){
    int a;
    int cnt=0,i=1;
    cin>>a;
    while (cnt!=a){
        i++;
        if (is_prime(i))cnt++;
    }
    cout<<i;
    return 0;
}