Codeforces842C dfs+set

题意：有一颗n个点，n-1条边的树，每个点的权值为wi，问从1到每个点，每条路径所经过点的gcd，可以选择将该条路径中某一个点改为0；
思路：dfs是肯定要的，至于set处理看代码吧；

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<map>
#include<stack>
#include<queue>
#include<algorithm>
#include<set>
#include<vector>
#include<cmath>
#define lmid l,m,rt<<1
#define rmid m+1,r,rt<<1|1
#define LL long long
#define rl r-l+1
using namespace std;
struct node
{
     int v,next;
}s[400005];
int head[200005];
int cnt=1;
void add(int u,int v)
{
     s[cnt].v=v;
     s[cnt].next=head[u];
     head[u]=cnt++;
}
int t[200005];
set<int>ans[200005];
int vis[200005];
int gcd(int a,int b)
{
     if(b==0)
          return a;
     return gcd(b,a%b);
}
void dfs(int v,int u,int d)
{
     for(int w:ans[u]) ans[v].insert(gcd(w,t[v]));
     ans[v].insert(d);
     d=gcd(d,t[v]);
     ans[v].insert(d);
     for(int i=head[v];~i;i=s[i].next)
     {
          int x=s[i].v;
          if(x!=u)
          {
               dfs(x,v,d);
          }
     }
}
int main()
{
     int n;
     scanf("%d",&n);
     for(int i=1;i<=n;i++)
          scanf("%d",&t[i]);
     memset(head,-1,sizeof(head));
     for(int i=1;i<=n-1;i++)
     {
          int u,v;
          scanf("%d%d",&u,&v);
          add(u,v);
          add(v,u);
     }
     dfs(1,0,0);
     for(int i=1;i<n;i++)
          cout<<*ans[i].rbegin()<<" ";
     cout<<*ans[n].rbegin()<<endl;
}