【杭电】[2196]Computer

文章字数:284

问题描述

问题分析

求树中所有点的最远距离

先找任意点的最远点u,在找u的最远点v,在这个过程中记录u到每个点的距离dis[0][i],再以v为起点寻找最远点,记录v到每个点的距离dis[1][i],则各个点的最远距离为max(dis[0][i],dis[1][i])

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include<stdio.h>
#include<string.h>
int n,m;
int t,max;
int dis[2][10200];
int head[10200];
int headcnt;
struct List {
	int u,v,w;
	int next;
} edge[20200];
int f(int a,int b) {
	return a>b?a:b;
}
void dfs(int u,int v,int w,int flag) {
	if(max<w)
		max=w,t=u;
	dis[flag][u]=w;
	for(int i=head[u]; i!=-1; i=edge[i].next) {
		if(edge[i].v!=v) {
			dfs(edge[i].v,u,w+edge[i].w,flag);
		}
	}
}
void add(int u,int v,int w) {
	edge[headcnt].u=u;
	edge[headcnt].v=v;
	edge[headcnt].w=w;
	edge[headcnt].next=head[u];
	head[u]=headcnt++;
}
int main() {
	while(scanf("%d",&n)!=EOF) {
		headcnt=0;
		memset(head,-1,sizeof(head));
		for(int i=2; i<=n; i++) {
			int v,w;
			scanf("%d %d",&v,&w);
			add(i,v,w);
			add(v,i,w);
		}
		max=0;
		dfs(1,0,0,0);
		max=0;
		dfs(t,0,0,0);
		dfs(t,0,0,1);
		for(int i=1; i<=n; i++) {
			printf("%d\n",f(dis[0][i],dis[1][i]));
		}
	}
	return 0;
}

该内容采用 CC BY-NC-SA 4.0 许可协议。

如果对您有帮助或存在意见建议,欢迎在下方评论交流。

加载中...