A7850.完成素数判定函数[题解]

有帮助就点个赞呗

#include<bits/stdc++.h>
using namespace std;
bool check(int a){
    for(int i=2;i<sqrt(a);i++){
        if(a%i==0)return false;
    }
    return true;
}
int main(){
    int n;
    cin>>n;
    if(n==1){
        cout<<"No";
        return 0;
    }else if(n==2){
        cout<<"Yes";
        return 0;
    }
    if(check(n)){
        cout<<"Yes";
    }else{
        cout<<"No";
    }
    return 0;
}