函数解题

#include <iostream>
using namespace std;
bool is_p(int x){
if(x<2){
return 0;
}
for(int i = 2;i<= x-1;i++){
if(x%i == 0){
return 0;
}
}
return 1;
}
int main(){
int n;
cin>>n;
if(is_p(n)){
cout<<"Yes";
}else{
cout<<"No";
}
return 0;
}