Codeforces 842C - Ilya And The Tree 【树上DFS】

**【题意】

给出一棵生成树，每个节点都有一个值，现在要求出每个节点的美丽值的最大值，美丽值的定义为从根节点到该节点(包含)路径上所有点的值的gcd，求解每个点时可以把路径上某一个点的值变为0。你可以认为每个点美丽值的求解是独立的。**

别人的代码ac。。

#include <bits/stdc++.h>
using namespace std;

const int maxn=2e5;

int n;
int A[maxn+10],dp[maxn+10][2],fa[maxn+10];
vector<int>g[maxn+10];
set<int>s[maxn+10];

void dfs(int v,int p,int b){
    for(int w:s[p])s[v].insert(__gcd(w,A[v]));
    s[v].insert(b);
    b=__gcd(b,A[v]);
    s[v].insert(b);
    for(int w:g[v])if(w!=p)dfs(w,v,b);
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",A+i);
    int u,v;
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1,0,0);
    for(int i=1;i<=n;i++)
        printf("%d ",*s[i].rbegin());
    return 0;
}

我的代码wa4

int val[N];
const int maxn=2e5;
int fst[N],nxt[N<<1],to[N<<1],e;
vector<int>g[maxn+10];
set<int>s[maxn+10];
set<int>::iterator it;
void dfs(int u,int fa,int c){
    for(it=s[u].begin();it!=s[u].end();++it)s[u].insert(__gcd(*it,val[u]));
    s[u].insert(c);
    c=__gcd(c,val[u]);
    s[u].insert(c);
    for(int i=0;i<g[u].size();++i)if(g[u][i]!=fa)dfs(g[u][i],u,c);
}
int main(){
    int n;sf("%d",&n);
    mem(fst,-1);e=0;
    rep(i,1,n)sf("%d",&val[i]);
    rep(i,1,n-1){ int u,v;sf("%d%d",&u,&v);g[u].push_back(v);g[v].push_back(u);}
    dfs(1,0,0);
    rep(i,1,n)pf("%d%c",*s[i].rbegin(),i==n?'\n':' ');

}

原本以为可以dp ，但其实是不可以的。。。谢谢提醒

int gcd(int a,int b){ if(b>a)swap(a,b);return b==0?a:gcd(b,a%b); }
int dp[N][2],ans[N];
void add(int u,int v){
    to[e]=v;nxt[e]=fst[u];fst[u]=e++;
}
void dfs(int u,int fa){
    dp[u][0]=gcd(dp[fa][0],val[u]);
    dp[u][1]=max(dp[fa][0],gcd(dp[fa][1],val[u]));
    ans[u]=max(dp[u][0],dp[u][1]);
    for(int i=fst[u];~i;i=nxt[i]){
        int v=to[i];
        if(v==fa)continue;
        dfs(v,u);
    }
}
int main(){
    int n;sf("%d",&n);
    mem(fst,-1);e=0;
    rep(i,1,n)sf("%d",&val[i]);
    mem(dp,0);
    rep(i,1,n-1){ int u,v;sf("%d%d",&u,&v);add(u,v);add(v,u); }
    dp[1][0]=val[1];
    dfs(1,1);
    //rep(i,1,n)pf("%d %d\n",dp[i][0],dp[i][1]);
    rep(i,1,n)pf("%d%c",ans[i],i==n?'\n':' ');
}