树论（二）

题目链接

• 很难维护两个集合中各取一个数能得到的gcd的最大值；事实上，相信你自己，就是“不可做”
• 转换视角，从倍数的角度出发，考虑从大到小枚举答案，任意边的答案只会被更新一次，用【并查集】维护已经被处理好的边将n个孤立的节点逐渐连接成树的过程，根节点代表了当前集合中还未被处理的那条边

点击查看代码

#include <bits/stdc++.h>
using namespace std;
vector<int>a[1000005];
int dep[1000005],n;
int f[1000005],id[1000005],fa[1000005],ans[1000005];
int get(int x)
{
	if(fa[x]==x)
	{
		return x;
	}
	return fa[x]=get(fa[x]);
}
void dfs(int n1)
{
	for(int i=0;i<a[n1].size();i++)
	{
		dep[a[n1][i]]=dep[n1]+1;
		dfs(a[n1][i]);
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n;
		for(int i=1;i<=n;i++)
		{
			a[i].clear();
			fa[i]=i;
		}
		for(int i=1;i<n;i++)
		{
			int u,v;
			cin>>u>>v;
			if(u>v)
			{
				swap(u,v); 
			}
			f[v]=u;
			id[v]=i;
			a[u].push_back(v);
		}
		dep[1]=1;
		dfs(1);
		for(int d=n/2;d>=1;d--)
		{
			int x=get(d);
			for(int i=2;i*d<=n;i++)
			{
				int y=get(i*d);
				while(x!=y)
				{
					if(dep[x]>dep[y])
					{
						swap(x,y);
					}
					ans[id[y]]=d;
					fa[y]=get(f[y]);
					y=fa[y];
				}
			}
		}
		for(int i=1;i<n;i++)
		{
			if(i>1)
			{
				cout<<' ';
			}
			cout<<ans[i];
		}
		cout<<endl;
	}
	return 0;
}