代码求调（输出都正确但是是0分）

#include <bits/stdc++.h>
using namespace std;

int yinzi(int a){
	int sum=0;
	for(int i=2;i*i<=a;i++){
		if(a%i==0){
			while(a%i==0){
				a/=i;
			}
			sum++; 
		}
		if(a==1){
			break;
		}
	}
	sum++;
	return sum;
}

int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		int a;
		cin>>a;
		if(yinzi(a)==2){
			cout<<1<<endl;
		}
		else cout<<0<<endl;
	}	
	return 0;
}