C++：判断输入的自然数是否为质数

代码如下：

#include <stdio.h>
int main() {

    int number;
    scanf("%d,&number");
    int flag = 0;
    int i;
    for(i = 2;i<number;i++){
        if(number % i ==0){
            flag = 1;
            break;
        }
    }
    if(flag == 1){
        printf("NO");
        //printf("a %d %d",i,flag);
    }else if(flag == 0){
        printf("YES");
        //printf("b %d %d",i,flag);
    }
    return 0;
}