蓝桥杯算法提高VIP-质数的后代 （C++代码）

#include<iostream>
using namespace std;
int zhishu(int b){
    if(b==1)return 0;
    int flag=0;
    for(int i=1;i*i<=b;i++){
        if(b%i==0)
        flag++;
    } 
    if(flag==1)return 1;
    else return 0;
} 
int main(){
    int T,a;
    cin>>T;
    while(T--){
        int flag;
        cin>>a;
            for(int i=2;i<a;i++){
                flag=0;
                if(zhishu(i)&&a%i==0){
                    if(zhishu(a/i)){
                        cout<<"Yes"<<endl;
                        flag=1;
                        break;
                    }
                } 
            }
            if(flag==0)cout<<"No"<<endl; 
    }
    return 0;
}