P10720 [GESP202406 五级] 小杨的幸运数字 欧拉筛

P10720 [GESP202406 五级] 小杨的幸运数字
prim[i]表示第i个质数
st[i]表示数字i的最小质因子

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+100;
int prim[N],vis[N],st[N],cnt = 0;
int n;
void init()
{
    n = 1e6;
    for(int i = 2 ; i <= n ; ++i)
    {
        if(!vis[i])prim[++cnt] = i,st[i] = i;
        for(int j = 1 ; 1ll * i * prim[j] <= n ; ++j)
        {
            vis[i*prim[j]] = 1;
            st[i*prim[j]] = prim[j];
            if(i%prim[j] == 0)break;
        }
    }
}
int main()
{
    init();
    cin>>n;
    for(int i = 1 ; i <= n ; ++i)
    {
        int x;cin>>x;
        int now = x;
        while(now%st[x] == 0)now/=st[x];
        if(now == 1){
            cout<<"0\n";continue;
        }
        x = now;
//         cout<<x<<" "<<st[x]<<endl;
        while(now%st[x] == 0)now/=st[x];
        if(now == 1)
            cout<<"1\n";
        else 
            cout<<"0\n";
//         cout<<now<<endl;
    }
    return 0;
}