[美团 CodeM 初赛 Round A]最长树链

题目大意：
　　给你一棵带点权的树，找出一个最长的树链满足链上点权的最大公因数不为1。

思路：
　　暴力DP。
　　对于每个点，记录一下以这个点为一个端点的所有链的最大公因数及长度。
　　然后暴力转移一下，时间复杂度O(n^2logn)，不过中间有一些情况是可以舍掉的，做不满。
　　本来是想试试这样暴力能拿多少分的，没想到直接A掉了。

#include<cstdio>
#include<cctype>
#include<vector>
#include<hash_map>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=100001;
int w[N],ans;
std::vector<int> e[N];
inline void add_edge(const int &u,const int &v) {
    e[u].push_back(v);
    e[v].push_back(u);
}
int gcd(const int &a,const int &b) {
    return b?gcd(b,a%b):a;
}
__gnu_cxx::hash_map<int,int> dfs(const int &x,const int &par) {
    __gnu_cxx::hash_map<int,int> map,tmp;
    map[w[x]]=1;
    for(unsigned i=0;i<e[x].size();i++) {
        const int &y=e[x][i];
        if(y==par) continue;
        tmp=dfs(y,x);
        for(register __gnu_cxx::hash_map<int,int>::iterator i=map.begin();i!=map.end();i++) {
            for(register __gnu_cxx::hash_map<int,int>::iterator j=tmp.begin();j!=tmp.end();j++) {
                if(gcd(i->first,j->first)!=1) {
                    ans=std::max(ans,i->second+j->second);
                }
            }
        }
        for(register __gnu_cxx::hash_map<int,int>::iterator i=tmp.begin();i!=tmp.end();i++) {
            const int tmp=gcd(w[x],i->first);
            if(tmp!=1) {
                map[tmp]=std::max(map[tmp],i->second+1);
            }
        }
    }
    return map;
}
int main() {
    const int n=getint();
    for(register int i=1;i<n;i++) {
        add_edge(getint(),getint());
    }
    for(register int i=1;i<=n;i++) {
        w[i]=getint();
    }
    dfs(1,0);
    printf("%d\n",ans);
    return 0;
}