（唯一分解定理，素数筛）CF1470B. Strange Definition

CF1470B. Strange Definition

题意：

定义如果两个数$x,y$满足$\frac{lcm(x,y)}{gcd(x,y)}$是一个完全平方数，则称$(x,y)$相邻。
给一个数列$a[1...n]$,定义$d_i$是和$a_i$相邻的元素的个数。定义数列的美丽程度为$\underset{1\le i\le n}{\max }{d}_i$。
每一秒钟过后，每个$a_i$会变成$\prod _{a_i和a_j相邻}{a}_j$,给你过去的时间，你要输出这么长时间过去后，数列的美丽程度。

思路：

由唯一分解定理可知，一个数可以表示成$x=p_1^{a_1}{p}_2^{a_2}\cdots p_n^{a_n}$。
$gcd(x,y)=p_1^{\min (a_{x,1},a_{y,1})}{p}_2^{\min (a_{x,2},a_{y,2})}\cdots p_n^{\min (a_{x,n},a_{y,n})}$
$lcm(x,y)=p_1^{\max (a_{x,1},a_{y,1})}{p}_2^{\max (a_{x,2},a_{y,2})}\cdots p_n^{\max (a_{x,n},a_{y,n})}$
可得$k=\frac{lcm(x,y)}{gcd(x,y)}=p_1^{|a_{x,1}-a_{y,1}|}{p}_2^{|a_{x,2}-a_{y,2}|}\cdots p_n^{|a_{x,n}-a_{y,n}|}$
可见，如果$k$是完全平方数，那么对于$i\in [1,n]，|a_{x_i}-a_{y,i}|$都是偶数。也就是对于两个数满足$a_{x,i}\equiv a_{y,i}(\mathrm{m}\mathrm{o}\mathrm{d}2)$。
这样我们令$x^{{}^{\prime }}=p_1^{a_1\mathrm{m}\mathrm{o}\mathrm{d}2}{p}_2^{a_2\mathrm{m}\mathrm{o}\mathrm{d}2}\cdots p_n^{a_n\mathrm{m}\mathrm{o}\mathrm{d}2}$,相同就表示他们相邻。
可以发现，时刻只有$0$和其他的区别，如果$0$时刻过后，有偶数个数是相邻的，那么相乘$x^{{}^{\prime }}=1$。
先预处理$1000$以内的素数，否则会T.

代码：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define int long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z)	for(x=y;x>=z;x--)
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=1e3+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
int prim[N],ok[N],pos=0;
void init()
{
	int n=N-1,i,j;
    for(i=2;i<=n;i++)
    {
        if(ok[i]==0)
            prim[++pos]=i;
        for(j=1;j<=pos;j++)
        {
            if(i*prim[j]>n)
                break;
            ok[i*prim[j]]=1;
            if(i%prim[j]==0)
                break;
        }
    }
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int t;
	cin>>t;
	init();
	while(t--)
	{
		int n,i,j,ans1=0;
		cin>>n;
		map<int,int> m;
		for(i=1;i<=n;i++)
		{
			int x;
			cin>>x;
			int tot=1;
			for(j=1;prim[j]<=x;j++)
			{		
				if(j>pos)
					break; 
				if(x%prim[j]==0)
				{
					int sum=0;
					while(x%prim[j]==0)
					{
						x/=prim[j];
						sum++;
					}
					if(sum%2==1)
						tot*=prim[j];
				}
			}
			if(x>1)
				tot*=x;
			m[tot]++;
			ans1=max(m[tot],ans1);
		}
		int ans2=0;
		for(auto it:m)
		{
			if(it.se%2==0 && it.fi!=1)
				m[1]+=it.se;
			it.se=0;
		}
		ans2=max(ans1,m[1]);
		int q;
		cin>>q;
		while(q--)
		{
			int w;
			cin>>w;
			if(w==0)
				cout<<ans1<<endl;
			else
				cout<<ans2<<endl;
		}
	}
	return 0;
}