【loj6159】「美团 CodeM 初赛 Round A」最长树链

题目描述
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;
}