题号 NC13233
名称 最长树链
来源 美团2017年CodeM大赛-初赛A轮
戳我进入往期每日一题汇总贴~
往期每日一题二期题单



如果你在题库做题时遇到了喜欢的题目，欢迎推荐给邓老师~ 点击查看详情
欢迎给每日一题投稿，投稿需要提供牛客题库里的题目+题解 投稿有牛币奖励，可发站内信给王清楚或联系QQ 234186389
每日一题QQ群:659028468

题解

作者：喵渺淼妙的死忠粉

思路:

首先我们处理出4e4以内的质数,然后因为任何合数都可以写成的形式,显然大于4e4的数,假如经过这么一次筛选,一定是不会出现超过4e4的合数.对于每个数,我们处理出来它们的质因子,这里呢,用的是最原始的判断方法,假如有n个数是大质数,那么就高达了.所以假如这样的话,最好还得套个米可罗宾先判断它是不是大质数.后面对于每个质因子搜其点,找到两条最长链即可了(随便树形dp一下).因为一个数的质因子个数最多不超过30个(1e9以内.)所以你的搜索上线也就.然后就得以解决此问题.

代码:

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5,M=1e4;
int prime[M],id=0,w[N];
vector<int>g[N];
map<int,vector<int> >s;
bool vis[N];
int ans=-1;
int dfs(int u,int num)
{
    vector<int>s;
    if(w[u]%num!=0)    return 0;
    vis[u]=true;int res=1;
    for(int x:g[u])
    {
        if(!vis[x])
        {
            int cnt=dfs(x,num);
            res=max(res,1+cnt);
            s.push_back(cnt);
        }
    }sort(s.begin(),s.end());
    int x=s.size();
    if(s.size()>1)    ans=max(ans,s[x-1]+s[x-2]+1);
    return res;
}

int main()
{
    int m=4e4;
    for(int i=2;i<=m;i++)
    {
        bool ok=true;    
        for(int j=2;j<=i/j;j++)
        {
            if(i%j==0)    ok=false;
        }if(ok)    prime[++id]=i;
    }int n;scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,v;scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&w[i]);int x=w[i];
        for(int j=1;j<=id;j++)
        {
            if(x==1)    break;
            if(x%prime[j]==0)
            {
                s[prime[j]].push_back(i);
                while(x%prime[j]==0)    x/=prime[j];
            }
        }if(x>1)    prime[++id]=x,s[x].push_back(i);
    }
    for(int i=1;i<=id;i++)
    {
        for(int j:s[prime[i]])
            if(!vis[j])    dfs(j,prime[i]);
        for(int j:s[prime[i]])    vis[j]=false;
    }printf("%d\n",ans);
    return 0;
}
/*
4
1 2
1 3
1 4
2 2 2 2 
*/

欢迎各位大佬来做题写题解，也欢迎大家踊跃在当日讨论贴中提问！

活动奖励：

在牛客博客中写出题解，并回复地址
审核通过可获得（依据题目难度和题解的内容而定）

本道题目1月14日中午12：00之前写的题解有获得牛币资格~

.牛币兑换中心

牛客博客开通方式

1. 如何开通牛客博客：https://www.nowcoder.com/discuss/202952
2. 如何使用博客搬家功能：进入博客--->设置--->底部博客搬家
3. 如果你对牛客博客有任何意见或建议：牛客博客意见反馈专贴