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

水一水

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

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