大佬们这哪里错了

没想好怎么改，这样还是不对#include<stdio.h>
int main()
{
int n,i,flag;
flag=0;
scanf("%d",&n);
for(i=2;i<n;i++)
{
if(n%i==0)
{
flag=1;
break;
}
}
if(flag==1)
{
printf("No");
}
else if(n=2)
{
printf("Yes");
}
else if(n=1)
{
printf("Yes");
}
else
{
printf("Yes");
}
return 0;
}