样例已过,但有些RE有些TLE,考试时候是AC的,不懂怎么回事,大佬求解,如果需要可以悬关
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6;
vector<bool> isprime(maxn+1,true);
vector<int> prime;
void xixish(int N)
{
isprime[0]=isprime[1]=false;
for(int i=2;i<=N;i++)
{
if(isprime[i]) prime.push_back(i);
for(int j=0;j<int(prime.size()) && prime[j]*i<=N;j++)
{
isprime[prime[j]*i]=false;
if(i%prime[j]==0) break;
}
}
}
int main()
{
int n,a[10001],mx=0;
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i],mx=max(mx,a[i]);
xixish(mx);
for(int i=1;i<=n;i++)
{
bool f=1;
int cnt=0,j=0;
while(a[i]>=prime[j])
{
if(a[i]%prime[j]==0) cnt++;
if(cnt==3){f=0;break;}
j++;
}
if(cnt==1) f=0;
cout<<f<<endl;
}
return 0;
}