C：键盘输入一个正整数，要求判断该数是否为素数

题目描述：
键盘输入一个正整数，要求判断该数是否为素数。素数即质数，只能被1和它本身整除。特别说明，1不是素数。如果是素数，输出“YES”，否则输出“NO”。

输入格式:
输入只有一个用例，即一个整数n。
输出格式:
如果n是素数，输出“YES”，否则输出“NO”。

题解：

#include<stdio.h>
int main()
{
    int m, n,flag=1;
    scanf("%d", &m);
    if(m==1){
    	printf("NO\n");
	}
	else {
		for (int i = 2; i < m; i++) {
		if (m % i == 0) {
			flag=0;
			break;
		}
	}
	if(flag==0)
        printf("NO\n");
    else
        printf("YES\n");
	}
    
}