求帮忙改下代码
一道判断素数的题,有些判断出来是错的#include <stdio.h>
#include <math.h>
int main( )
{
int ri, repeat;
int flag, i, m, n;
scanf("%d", &repeat);
for(ri=1; ri<=repeat; ri++){
scanf("%d", &m);
if(m==1)
flag=0;
else if(m==2)
flag=1;
for(i=2;i<m/2;i++)
{
if(m%i==0)
{
flag=0;
break;
}
else flag=1;
if(flag) printf("YES\n");
else printf("NO\n");
}
}
}