PTA 7-24 求素数

本题目要求读入1个正整数A，判断A是否为素数。

小贴士：反过来考虑，只要有一次整除了，就不是素数，循环break，不要继续算了。还可以加入标志法，flag的初值为1，一,旦整除了，flag置0。输出结果根据flag的值是0还是1来判断。当然也可不用标志法。

输入格式:

输入1个正整数A。

输出格式:

输出A是或否为素数。

输入样例:

2

输出样例:

2 yes

#include 
int main(){
    int A;
    int flag = 0;
    scanf("%d", &A);
    if(A == 1 || A <= 0){
        printf("%d no", A);
    }else{
        for(int i = 2; i <= A - 1; i++){
            if(A % i == 0) 
                flag = 1;
        }
        if(flag == 0)
            printf("%d yes", A);
        else
            printf("%d no", A);
    }
    return 0;
}