D. GCD Counting(树上dp)

题目链接:http://codeforces.com/contest/1101/problem/D

题目大意:给你n个点,每个点都有权值,然后给你边的关系,问你树上的最大距离。(这里的最大距离指的是这条路径上的所有数最大gcd>1)。

 具体思路:首先,我们可以找出对于一个父亲节点,他和子节点能够在不是互素的前提下,对于他们每一个因子的个数。

然后我们先dfs到底部,然后看他的每一个子节点之间产生的最大公因数是不是1,如果不是的话,我们就求这条路径上的最大值,在求最大值的时候,我们还是按照树的结构来,如图所示,从4号节点开始,4号节点更新之后会回到上一层,也就是3号节点,然后再求3号节点的所有因子和4号节点的因子是不是互素的,如果不是就从不是互素的这些状态里面找出一个最大值,然后在更新3号节点,这样更新下去就可以了。

注意:应该是先求最大状态再去更新父亲节点,如果先更新再去找最大值的话,会有一部分路径是重复的。举个例子,对于2,3,6这三个点,正确的解法是先求2和当前3号节点的最大距离,然后更新2号节点,然后再轮到6号节点,这个时候2号节点里面存储的是和3号节点之间的关系,我们再去求和6号节点之间的关系,这样就不会有路径重复了。

 

AC代码:

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 # define ll long long
 4 const int maxn = 4e5+100;
 5 int a[maxn],num,head[maxn],maxx;
 6 map<int,int>dis[maxn];
 7 struct node
 8 {
 9     int nex;
10     int to;
11 } edge[maxn];
12 void init()
13 {
14     maxx=0;
15     num=0;
16     memset(head,-1,sizeof(head));
17 }
18 int gcd(int n,int m)
19 {
20     if(n<m)
21         swap(n,m);
22     return n%m==0?m:gcd(m,n%m);
23 }
24 void addedge(int fr,int to)
25 {
26     edge[num].nex=head[fr];
27     edge[num].to=to;
28     head[fr]=num++;
29 }
30 void dfs(int fr,int rt)
31 {
32     if(a[fr]>1)
33     {
34         dis[fr][a[fr]]=1;
35     }
36     for(int i=head[fr]; i!=-1; i=edge[i].nex)
37     {
38         int to=edge[i].to;
39         if(to==rt)continue;
40             dfs(to,fr);
41         for(auto t1:dis[fr])
42         {
43             for(auto t2:dis[to])
44             {
45                 if(gcd(t1.first,t2.first)!=1)
46                 {
47                     maxx=max(maxx,t1.second+t2.second);
48                 }
49             }
50         }
51         for(auto t1:dis[to])
52         {
53             int tmp=gcd(a[fr],t1.first);
54             if(tmp==1)
55                 continue;
56             dis[fr][tmp]=max(dis[fr][tmp],t1.second+1);
57         }
58     }
59 }
60 int main()
61 {
62     init();
63     int n;
64     scanf("%d",&n);
65     int t1,t2;
66     for(int i=1; i<=n; i++)
67     {
68         scanf("%d",&a[i]);
69         if(a[i]!=1)
70             maxx=1;
71     }
72     for(int i=1; i<=n-1; i++)
73     {
74         scanf("%d %d",&t1,&t2);
75         addedge(t1,t2);
76         addedge(t2,t1);
77     }
78     dfs(1,1);
79     printf("%d\n",maxx);
80     return 0;
81 }

 

posted @ 2019-01-12 15:13  Let_Life_Stop  阅读(668)  评论(0编辑  收藏  举报