鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=2196
題意:有n臺電腦,通過n-1條網線連接,問每臺電腦到離他最遠的距離是多少。
分析:我們把這些電腦看成一棵樹,可以很容易的得到每個點的子樹最遠的距離(a),我們發現某個節點的答案就是max(a,除下它所在子樹後父親最遠的距離)。
所以我們可以先dfs一遍存下每個點(u)的子樹離他最遠和次遠距離,然後我們可以在第二次dfs時判斷它是不是更新它父親(fa)最遠的點,是的話max(a,父親次遠),否max(a,父親最遠),然後把u看成父親,fa看成兒子更新u的最遠次遠距離。
代碼:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 10010
#define Mm 20020
#define mod 1000000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
struct edge {
int v,w,next;
}e[Mm];
int tot,head[Mn];
void addedge(int u,int v,int w) {
e[tot].v=v;
e[tot].w=w;
e[tot].next=head[u];
head[u]=tot++;
}
int dp[Mn][2];
void dfs(int u,int fa) {
dp[u][0]=0;
dp[u][1]=0;
for(int i=head[u];~i;i=e[i].next) {
int v=e[i].v;
int w=e[i].w;
if(v==fa) continue;
dfs(v,u);
if(dp[v][0]+w>=dp[u][0]) {
dp[u][1]=dp[u][0];
dp[u][0]=dp[v][0]+w;
} else if(dp[v][0]+w>dp[u][1]){
dp[u][1]=dp[v][0]+w;
}
}
}
void dfs2(int u,int fa) {
for(int i=head[u];~i;i=e[i].next) {
int v=e[i].v;
int w=e[i].w;
if(v==fa) continue;
if(dp[v][0]==dp[u][0]-w) {
if(dp[u][1]+w>=dp[v][0]) {
dp[v][1]=dp[v][0];
dp[v][0]=dp[u][1]+w;
} else if(dp[u][1]+w>dp[v][1]) {
dp[v][1]=dp[u][1]+w;
}
} else {
if(dp[u][0]+w>=dp[v][0]) {
dp[v][1]=dp[v][0];
dp[v][0]=dp[u][0]+w;
} else if(dp[u][0]+w>dp[v][1]){
dp[v][1]=dp[u][0]+w;
}
}
dfs2(v,u);
}
}
int n;
void init() {
CLR(head,-1);
tot=0;
}
int main() {
while(~scanf("%d",&n)) {
init();
for(int i=2;i<=n;i++) {
int u,w;
scanf("%d%d",&u,&w);
addedge(i,u,w);
addedge(u,i,w);
}
dfs(1,0);
dfs2(1,0);
for(int i=1;i<=n;i++) {
printf("%d\n",dp[i][0]);
}
}
return 0;
}