判断输入的数字是否是质数

//输入一个数，判断是否是质数

#import <Foundation/Foundation.h>

int main(int argc, const char * argv[]) {
    @autoreleasepool {
        
                int a ;
                scanf("%d",&a);
                int i ;
                if (a == 0 || a == 1) {
                    printf("输入的不是质数 \n");
                }
            else
                {
               for (i = 2; i <= a; i++) {
        
                    if ((i != 1 && i != a && a % i == 0 )) //a=1 a= 0 OR 除了a本身和1之外，存在一个能被a整除的i
                    {
                        printf("输入的不是质数 \n");
                        return 0;
                    }
                        }
                    printf("输入的是质数\n");
                }
    }
    return 0;
}