计蒜客-判断质数-C语言

C代码如下：
以下代码均在计蒜客平台提交编译通过。

#include<stdio.h>
int main()
{
    int a=0,i,j,b=0;
    int flag=0;
    scanf("%d",&a);
    if(a==1)
        printf("NO");
    for(i=0;i*i<=a;i++)
    {
        for(j=2;j<=i;j++)
        {
            if(a%j==0)
            {
                flag=1;

            }

        }

    }
    if(flag==1)
        printf("NO");
    if(flag==0&&a!=1)
        printf("YES");
    return 0;
}