codeforces 842C. Ilya And The Tree(dfs)

题意：

有n个节点标号1~n，每个节点有一个正整数价值，这几个节点形成了一棵以节点1为根节点的树，求根节点到节点x的路径上所有节点价值的gcd（你可改变这条路径上某个节点的的价值为0，或者不做任何修改）；

思路：

从根节点开始，求每个节点可能的gcd，用set维护这些gcd。则每个节点的可能的gcd必包括 其父节点的gcd与当前节点的价值的最大公因数。但漏了一种情况，不修改从根节点到当前节点的根节点的任何价值，只修改当前节点的价值，所以要维护一个从根节点到当前节点的根节点的gcd。

代码：

#include <bits/stdc++.h>

using namespace std;
const int maxn  = 2e5+7;
set<int> s[maxn];
int arr[maxn];
vector<int> e[maxn];
set<int>::iterator it;
void dfs(int u,int fa,int gcd)
{
for(it=s[fa].begin();it!=s[fa].end();it++)
s[u].insert(__gcd(*it,arr[u]));
s[u].insert(gcd);
gcd = __gcd(gcd,arr[u]);
s[u].insert(gcd);
int len = e[u].size();
for(int i = 0;i<len;i++)
{
int v = e[u][i];
if(v!=fa)
{
dfs(v,u,gcd);
}
}
}
int main()
{
int n;
cin>>n;
for(int i = 1;i<=n;i++)
scanf("%d",arr+i);
for(int i = 0;i<n-1;i++)
{
int u,v;
scanf("%d%d",&u,&v);
e[u].push_back(v);
e[v].push_back(u);
}
dfs(1,0,0);
for(int i = 1;i<=n;i++)
{
printf("%d ",*s[i].rbegin());
}
return 0;
}