題目描述
Mr. Walker\text{Mr. Walker}Mr. Walker最近在研究樹，尤其是最長樹鏈問題。現在樹中的每個點都有一個值，他想在樹中找出最長的鏈，使得這條鏈上對應點的值的最大公約數不等於111。請求出這條最長的樹鏈的長度。
輸入格式
第一行一個整數 nnn，表示點的個數。

接下來n−1n-1n−1行，每行兩個整數x,yx,yx,y表示x,yx,yx,y之間有邊。

資料保證給出的是一棵樹。

接下來一行nnn個整數表示每個點對應的權值 aia_ia
​i
​​
輸出格式
輸出一個整數，表示這條樹鏈的長度。
樣例
樣例輸入

4
1 2
1 3
2 4
6 4 5 2
樣例輸出

3
資料範圍與提示
1≤n≤1000001 \leq n \leq 1000001≤n≤100000

1≤ai≤1091 \leq a_i \leq 10^91≤a
​i
​​ ≤10
​9
​​

題解
列舉所有包含的質因子，把含有這個因子的點取出來，求直徑即可。

程式碼

#include<bits/stdc++.h>
#define ll long long
#define inf 1000000000
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'
 
){if (ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
map<int,int>mp;
int tot,ans,N;
vector<int>l[100005],g[100005];
int n,dis[100005];
bool flag[100005],vis[100005],mark[100005];
int bfs(int u)
{
    queue<int>q;int ans=0;
    q.push(u);vis[u]=1
 
;dis[u]=1;
    while (!q.empty())
    {
        int now=q.front();q.pop();
        for (int i=0;i<l[now].size();i++)
        {
            if (!mark[l[now][i]]) continue;
            if (vis[l[now][i]]) continue;
            dis[l[now][i]]=dis[now]+1;
            if (dis[l[now][i]]>dis[ans]) ans=l[now][i];
            q.push(l[now][i]);vis[l[now][i]]=1;
        }
    }
    q.push(ans);flag[ans]=1;dis[ans]=1;
    while (!q.empty())
    {
        int now=q.front();q.pop();
        for (int i=0;i<l[now].size();i++)
        {
            if (!mark[l[now][i]]) continue;
            if (flag[l[now][i]]) continue;
            dis[l[now][i]]=dis[now]+1;
            if (dis[l[now][i]]>dis[ans]) ans=l[now][i];
            q.push(l[now][i]);flag[now]=1;
        }
    }
    return dis[ans];
}
int main()
{
    n=read();
    for (int i=1;i<n;i++)
    {
        int x=read(),y=read();
        l[x].push_back(y);
        l[y].push_back(x);
    }
    for (int i=1;i<=n;i++)
    {
        int x=read();
        for (int j=2;j*j<=x;j++)if (x%j==0)
        {
            g[mp[j]?mp[j]:(mp[j]=++N)].push_back(i);
            while (x%j==0) x/=j;
            if (x==1) break;
        }
        if (x!=1) g[mp[x]?mp[x]:(mp[x]=++N)].push_back(i);
    }
    for (int i=1;i<=N;i++)
    {
        for (int j=0;j<g[i].size();j++)
        {
            mark[g[i][j]]=1;
            vis[g[i][j]]=0;
            flag[g[i][j]]=0;
        }
        for (int j=0;j<g[i].size();j++)if (!vis[g[i][j]]) ans=max(ans,bfs(g[i][j]));
        for (int j=0;j<g[i].size();j++)
        {
            mark[g[i][j]]=0;
            vis[g[i][j]]=1;
            flag[g[i][j]]=1;
        }
    }
    cout<<ans;
    return 0;
}