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

水一水

枚举各个质数，把是这个数倍数的点留下，跑直径，没了

#include <bits/stdc++.h>
using namespace std;
int h,t,n,p,q,M,N;
int dis[200001],lis[200001],fir[200001],nex[400001],to[400001];
vector<int> po[200001];
map<int,int> ma;
bitset<200001> vis,ok,v;
void add(int p,int q)
{
    to[++M]=q;nex[M]=fir[p];fir[p]=M;
}
int bfs(int now)
{
    dis[now]=1;vis[now]=1;int lo=0,id=0;
    v.reset();v[now]=1;
    for(lis[h=t=1]=now;h<=t;h++)
        for(int i=fir[lis[h]];i;i=nex[i])
        if(ok[to[i]] && !v[to[i]])
        {
            lis[++t]=to[i];
            vis[to[i]]=1;v[to[i]]=1;
            dis[to[i]]=dis[lis[h]]+1;
            if(dis[to[i]]>lo) lo=dis[to[i]],id=to[i];
        }
    return id;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
        scanf("%d%d",&p,&q),add(p,q),add(q,p);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&t);
        for(int j=2;j<=sqrt(t);j++)
        {
            if(t%j==0)
                po[ma[j]?ma[j]:(ma[j]=++N)].push_back(i);
            while(t%j==0)
                t/=j;
        }
        if(t>1) po[ma[t]?ma[t]:(ma[t]=++N)].push_back(i); 
    }
    int L=0;
    for(int i=1;i<=N;i++)
    {
        ok.reset();vis.reset();
        for(int j=0;j<po[i].size();j++)
            ok[po[i][j]]=1;//,printf("%d ",po[i][j]);
    //    puts("");
        for(int j=0;j<po[i].size();j++)
        if(!vis[po[i][j]])
            L=max(L,dis[bfs(bfs(po[i][j]))]);
    }
    printf("%d
",L);
    return 0;
}