P7854 「EZEC-9」GCD Tree

题目

题目

思路

牛马题终于A了交了至少30发
首先可以去重，把所有权值相同的点搞成一条链，如果要加个儿子就加在最后面的点上
显然有的是有解时的树的形状：
每个点的父亲是其所有出现的因子中最大的
所以有一个结论：
去重后的树高最大$logn$
接下来发现一些用来判-1的性质：
首先，如果所有的gcd没出现，显然-1
接下来，如果某个数有2个因子都出现了，而且这2个因子直接不是直接的祖先关系，就是意味着这个数既要在这个因子下面，又要在那个因子下面，而这两个因子又没有共同的子树，所以这玩意没得放，-1
最后，考虑2个数的lca与其gcd不一样，-1。这意味着如果这破树合法，考虑每个点与其父亲的商$a_i$，对于某个$d$，所有$d$的倍数的$a_i$都得在一条链上。
根据这些判就行。
code:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int mod=1e9+7;
inline int read()
{
    int ret,c,f=1;
    while (((c=getchar())> '9'||c< '0')&&c!='-');
    if (c=='-') f=-1,ret=0;
    else ret=c-'0';
    while ((c=getchar())>='0'&&c<='9') ret=(ret<<1)+(ret<<3)+c-'0';
    return ret*f;
}
int n,b[1000006],f[1000006],siz[1000006],d[1000006],op[1000006],dep[1000006],a[1000006],p[1000006],tot;
vector<int> u[1000006];
int fir[1000006];
bool cmp(int x,int y)
{
	return b[x]<b[y];
}
bool comp(int x,int y)
{
	return dep[x]<dep[y];
}
int check(int x)
{
	int s=0;
	while (x!=0) s+=siz[x],x=fir[f[x]];
	return s;
}
int main()
{
	n=read();
	for (int i=1;i<=n;i++) op[i]=i,b[i]=read();
	int x=b[1];
	sort(op+1,op+1+n,cmp);
	for (int u=1;u<=n;u++)
	{
		int i=op[u],op=1;
		x=__gcd(x,b[i]);
		if (p[b[i]]!=0)
		{
			fir[i]=fir[p[b[i]]];
			f[i]=p[b[i]];
			dep[i]=dep[f[i]]+1;
			p[b[i]]=i;
			continue;
		}
		p[b[i]]=i;
		fir[i]=i; 
		if (b[i]==1)
		{
			continue;
		}
		for (int j=2;j*j<=b[i];j++)
		{
			if (b[i]%j==0)
			{
				if (j*j!=b[i])
				{
					if (p[b[i]/j]!=0&&p[j]!=0&&b[i]/j%j!=0)
					{
						cout<<-1;
						return 0;
					}
				}
			}
		}
		for (int j=2;j*j<=b[i];j++)
		{
			if (b[i]%j==0)
			{
				if (j*j!=b[i])
				{
					if (p[b[i]/j]!=0)
					{
						op=max(op,b[i]/j);
					}
					if (p[j]!=0)
					{
						op=max(op,j);
					}
				}
				else
				{
					if (p[j]!=0)
					{
						op=max(op,j);
					}
				}
			}
		}
		f[i]=p[op];
		dep[i]=dep[f[i]]+1;
	}
	if (p[x]==0)
	{
		cout<<-1;
		return 0;
	}
	b[0]=1;
	for (int i=1;i<=n;i++)
	{
		a[i]=b[i]/b[f[fir[i]]];
		if (i==fir[i]) u[a[i]].push_back(fir[i]);
	}
	for (int i=2;i<=1000000;i++)
	{
		int tmp=0;
		for (int k=i;k<=1000000;k+=i) for (int j=0;j<u[k].size();j++) op[++tmp]=u[k][j]; 
		sort(op+1,op+tmp+1,comp);
		for (int j=1;j<=tmp;j++)
		{
			if (check(op[j])!=j-1)
			{
				cout<<-1;
				return 0;
			}
			siz[op[j]]++;
		}
		for (int j=1;j<=tmp;j++)
		{
			siz[op[j]]--;
		}
	}
	for (int i=1;i<=n;i++) printf("%d ",f[i]);
	return 0;
}