Codeforces842 C. Ilya And The Tree（树形dp，思维，因子分解）

题意：

给定n个点的树，每个点有一个点权，点1是根节点。
每个点的答案是根到这个点的gcd。

现在你最多可以删除一个点的点权，问每个点的最大答案是多少。
对于每个点的询问是独立的。

数据范围：n<=2e5，1<=a(i)<=2e5

解法：

因为每个点的答案是根到这个点的gcd，那么每个点都和根有关，
因此考虑两种情况：
1.删掉根的点权。
2.不删掉根的点权。

1.对于情况1，每个点的答案就是每个点的路径gcd。
2.对于情况2，每个点的答案一定是根节点点权的因子，先O(sq)分解出根节点的因子，
在dfs向下搜索的过程中：
令cnt[j]表示对于当前节点，根节点到当前节点的路径上，有多少个数是第j个因子的倍数，
如果cnt[j]>当前点到根的链长度-1，那么说明删除一个点的点权可以使得每个数都是第j个因子的倍数，
用这个因子更新当前点的答案。

code：

#include<bits/stdc++.h>
using namespace std;
const int maxm=2e5+5;
vector<int>g[maxm];
vector<int>fac;
int ans1[maxm];//不选根
int ans2[maxm];//选根
int cnt[maxm];
int dep[maxm];
int a[maxm];
int n;
void dfs(int x,int fa){
    for(int v:g[x]){
        if(v==fa)continue;
        dep[v]=dep[x]+1;
        ans1[v]=__gcd(ans1[x],a[v]);
        for(int j=0;j<(int)fac.size();j++){
            if(a[v]%fac[j]==0)cnt[j]++;
            if(cnt[j]>=dep[x]-1)ans2[v]=max(ans2[v],fac[j]);//-1是因为可以删掉一个点权
        }
        dfs(v,x);
        for(int j=0;j<(int)fac.size();j++){
            if(a[v]%fac[j]==0)cnt[j]--;
        }
    }
}
signed main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    for(int i=1;i*i<=a[1];i++){
        if(a[1]%i==0){
            fac.push_back(i);
            if(a[1]/i!=i){
                fac.push_back(a[1]/i);
            }
        }
    }
    dep[1]=1;
    dfs(1,1);
    ans2[1]=a[1];
    for(int i=1;i<=n;i++){
        cout<<max(ans1[i],ans2[i])<<' ';
    }
    return 0;
}

---