C语言-用函数判断一个数是不是质数（素数）

注意：1不是质数（素数）；
质数的定义是： 只能够被1和本身整除的数

#include <stdio.h>
int is_prime(int n){
     if(n==1 ||n==0)
         return 1;
    for (int i = 2; i<n; i++) {
        if (n%i==0) {
            return 1;
        }
    }
    return 0;
}

int main(){
    int n;
    scanf("%d",&n);
    if (is_prime(n)) {
        printf("NO");
    }
    else
        printf("YES");
    return 0;
}