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

水一水

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

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 int h,t,n,p,q,M,N;
 4 int dis[200001],lis[200001],fir[200001],nex[400001],to[400001];
 5 vector<int> po[200001];
 6 map<int,int> ma;
 7 bitset<200001> vis,ok,v;
 8 void add(int p,int q)
 9 {
10     to[++M]=q;nex[M]=fir[p];fir[p]=M;
11 }
12 int bfs(int now)
13 {
14     dis[now]=1;vis[now]=1;int lo=0,id=0;
15     v.reset();v[now]=1;
16     for(lis[h=t=1]=now;h<=t;h++)
17         for(int i=fir[lis[h]];i;i=nex[i])
18         if(ok[to[i]] && !v[to[i]])
19         {
20             lis[++t]=to[i];
21             vis[to[i]]=1;v[to[i]]=1;
22             dis[to[i]]=dis[lis[h]]+1;
23             if(dis[to[i]]>lo) lo=dis[to[i]],id=to[i];
24         }
25     return id;
26 }
27 int main()
28 {
29     scanf("%d",&n);
30     for(int i=1;i<n;i++)
31         scanf("%d%d",&p,&q),add(p,q),add(q,p);
32     for(int i=1;i<=n;i++)
33     {
34         scanf("%d",&t);
35         for(int j=2;j<=sqrt(t);j++)
36         {
37             if(t%j==0)
38                 po[ma[j]?ma[j]:(ma[j]=++N)].push_back(i);
39             while(t%j==0)
40                 t/=j;
41         }
42         if(t>1) po[ma[t]?ma[t]:(ma[t]=++N)].push_back(i); 
43     }
44     int L=0;
45     for(int i=1;i<=N;i++)
46     {
47         ok.reset();vis.reset();
48         for(int j=0;j<po[i].size();j++)
49             ok[po[i][j]]=1;//,printf("%d ",po[i][j]);
50     //    puts("");
51         for(int j=0;j<po[i].size();j++)
52         if(!vis[po[i][j]])
53             L=max(L,dis[bfs(bfs(po[i][j]))]);
54     }
55     printf("%d\n",L);
56     return 0;
57 } 

 

posted @ 2017-07-31 09:38  汪立超  阅读(162)  评论(0编辑  收藏