从大到小枚举gcd的值
d
d
d,以及编号为
d
d
d的倍数的点,
[
d
,
2
d
,
3
d
,
…
]
[d,2d,3d,\dots]
[d,2d,3d,…]。
然后对于任何一条边
(
x
,
y
)
(x,y)
(x,y),如果
x
x
x的子树和
y
y
y的子树里都有编号为
d
d
d倍数的点,则这条边的答案至少为d。考虑到对于每条边我们只需要知道最大值,所以如果一条边已经在之前的
d
d
d中被更新过答案,我们就可以将它合并起来。合并的过程可以通过并查集来实现。
所以总结下来做法就是枚举出编号为 d d d的倍数的点之后,将这些点之间的路径都遍历一遍并合并起来。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int t,n,f[maxn];
int eu[maxn],ev[maxn];
inline int find(int x){
return f[x]==x?f[x]:f[x]=find(f[x]);
}
vector<int> g[maxn];
int par[maxn],dep[maxn];
void dfs(int u,int fa){
par[u]=fa;dep[u]=dep[fa]+1;
for(auto v:g[u]){
if(v==fa)continue;
dfs(v,u);
}
}
int ind[maxn],ans[maxn];
signed main(){
int size(256<<20); //256M
__asm__ ( "movq %0, %%rsp\n"::"r"((char*)malloc(size)+size));
ios::sync_with_stdio(0);
cin.tie(0);
//freopen("5.in","r",stdin);
//freopen("5.out","w",stdout);
cin>>t;
while(t--){
cin>>n;
for(int i=1;i<=n;i++)g[i].clear();
for(int i=1;i<n;i++){
cin>>eu[i]>>ev[i];
g[eu[i]].push_back(ev[i]);
g[ev[i]].push_back(eu[i]);
}
dfs(1,0);
for(int i=1;i<n;i++){
if(dep[eu[i]]>dep[ev[i]]){
ind[eu[i]]=i;
}
else{
ind[ev[i]]=i;
}
}
for(int i=1;i<=n;i++)f[i]=i;
for(int d=n/2;d>=1;d--){
int x=find(d);
for(int j=d+d;j<=n;j+=d){
int y=find(j);
while(x!=y){
if(dep[x]>dep[y])swap(x,y);
ans[ind[y]]=d;
f[y]=find(par[y]);
y=find(par[y]);
}
}
}
for(int i=1;i<n;i++)printf("%d ",ans[i]);
puts("");
}
exit(0);
//return 0;
}
每条边只会被合并一次,然后枚举倍数的时间开销也是调和级数,所以总复杂度为 O ( n log n ) O(n\log n) O(nlogn)。