POJ1985 Cow Marathon （树的直径）

用两次dfs求出树的直径，这两次dfs可以写在一起，当然为了方便理解，这里是分开写的。

 1 //两次dfs求树的重心
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 using namespace std;
 6
 7 const int N=40005,M=40005*2;
 8 int n,m,tot,p,ed;
 9 int d[N],head[N],to[M],w[M],nxt[M];
10
11 void add(int x,int y,int z){
12     nxt[++tot]=head[x];
13     head[x]=tot;
14     to[tot]=y;
15     w[tot]=z;
16 }
17
18 void dfs1(int u,int fa){
19     if(d[u]>d[p]) p=u;//更新从起点1能走到的最远点p
20     for(int i=head[u];i;i=nxt[i]){
21         int v=to[i];
22         if(v==fa) continue;
23         d[v]=d[u]+w[i];
24         dfs1(v,u);
25     }
26 }
27
28 void dfs2(int u,int fa){
29     if(d[u]>d[ed]) ed=u;
30     for(int i=head[u];i;i=nxt[i]){
31         int v=to[i];
32         if(v==fa) continue;
33         d[v]=d[u]+w[i];
34         dfs2(v,u);
35     }
36 }
37
38 int main(){
39     scanf("%d%d",&n,&m);
40     for(int i=1;i<=m;i++){
41         int x,y,z;
42         char d;
43         scanf("%d%d%d %c",&x,&y,&z,&d);
44         add(x,y,z);add(y,x,z);
45     }
46     dfs1(1,0);
47     memset(d,0,sizeof(d));
48     dfs2(p,0);
49     printf("%d\n",d[ed]);
50     return 0;
51 }