PTA循环结构 判断素数

输入一个正整数m，如果它是素数，输出"YES"，否则，输出"NO"（素数就是只能被1和自身整除的正整数，1不是素数，2是素数）。

输入格式:

在一行中输入一个正整数m。

输出格式:

对每一组输入，在一行中输出结果并换行。

输入样例:

17

输出样例:

YES

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