记得给1做特判（题解）

#include<iostream>
using namespace std;
int is_prime(int n)
{
    if(n==1||n==0)
    {
        return false;
    }
    for(int i=2;i<=n/2;i++)
    {
        if(n%i==0)
        {
            return false;
        }
    }
    return true;
}
int main(){
	int n;
	cin>>n;
	if(is_prime(n))
    {
        cout<<"Yes";
    }
	else
    {
        cout<<"No";
    }
}