hdu4607 树直径（两次dfs求向下最远距离及另一方向最远距离相加）

之前有想过树的直径咋求，不过一直没仔细去考虑。

昨天做了hdu2196 今天突然想到如果把max(up[i]+dis1[i])求出来不就是直径maxx嘛==

随后百度了一下看到别人大多数是两次bfs，我从hdu上找了一道题目，果然是可以的。

思想应该是一样的吧，都是求两条最长路。

这题用到最长路还有：

1.k<=maxx+1那么就是k-1

2.k>maxx+1就是(k-maxx-1)*2+maxx+1因为要来回（thinking）

 1 #include<stdio.h>
 2 #include<string.h>
 3 #include<algorithm>
 4 using namespace std;
 5 int next[200005],head[200005],point[200005],value[200005],now;
 6 int dis1[200005],dis2[200005],id[200005],up[200005];
 7 void add(int x,int y,int w)
 8 {
 9   next[++now]=head[x];
10   head[x]=now;
11   point[now]=y;
12   value[now]=w;
13 }
14 void dfs1(int pre,int u)
15 {
16   int i,v,w;
17   dis1[u]=dis2[u]=id[u]=0;
18   for (i=head[u];i;i=next[i])
19   {
20     v=point[i]; w=value[i];
21     if (v==pre) continue;
22     dfs1(u,v);
23     if (dis1[v]+w>dis1[u]){
24       dis1[u]=dis1[v]+w;
25       id[u]=v;
26     }
27   }
28   for (i=head[u];i;i=next[i])
29   {
30     v=point[i]; w=value[i];
31     if (v==pre||v==id[u]) continue;
32     if (dis1[v]+w>dis2[u]) dis2[u]=dis1[v]+w;
33   }
34 }
35 void dfs2(int pre,int u)
36 {
37   int i,v,w;
38   for (i=head[u];i;i=next[i])
39   {
40     v=point[i]; w=value[i];
41     if (v==pre) continue;
42     up[v]=up[u]+w;
43     if (id[u]!=v) up[v]=max(up[v],dis1[u]+w);
44     else up[v]=max(up[v],dis2[u]+w);
45     dfs2(u,v);
46   }
47 }
48 int main()
49 {
50   int n,i,x,y,m,k,maxx,T;
51   scanf("%d",&T);
52   while (T--)
53   {
54     scanf("%d%d",&n,&m);
55     memset(head,0,sizeof(head));
56     now=0;
57     for (i=2;i<=n;i++)
58     {
59       scanf("%d%d",&x,&y);
60       add(x,y,1); add(y,x,1);
61     }
62     memset(dis1,0,sizeof(dis1));
63     memset(dis2,0,sizeof(dis2));
64     memset(id,0,sizeof(id));
65     dfs1(0,1);
66     memset(up,0,sizeof(up));
67     dfs2(0,1);
68     maxx=0;
69     for (i=1;i<=n;i++)
70       maxx=max(maxx,up[i]+dis1[i]);
71     while (m--)
72     {
73       scanf("%d",&k);
74       if (k<=maxx+1) printf("%d\n",k-1);
75       else printf("%d\n",(k-maxx-1)*2+maxx);
76     }
77   }
78   return 0;
79 }

View Code

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4607

转载于:https://www.cnblogs.com/xiao-xin/articles/4234064.html