題目鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=2196
題目意思:求每個點在這棵樹中的最長距離
思路:從一個點找最短路徑,要麼往兒子找,要麼往父親找,這就出現了兩個遞推方向。
1.所以從根節點開始往下遍歷,找出每個點往下找的最長距離,記錄下來dp[x][0]。這裏順便記錄下次長距離,下一步用dp[x][1]。
2.往父親方向找:如果發現父親開始的最長距離是經過自己的,那麼就不能選這條路,
所以選擇次長路dp[e.to][2]=max(dp[x][1]+e.w,dp[x][2]+e.w)。
否則就選最長路:dp[e.to][2]=max(dp[x][0]+e.w,dp[x][2]+e.w)
代碼:
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
typedef long long LL;
typedef pair<int, int> P;
const int maxn = 1e4 + 5;
const int mod = 1e9 + 7;
int n;
struct edge{
int to,w;
};
vector<edge>G[maxn];
int dp[maxn][3];
int nxt[maxn];
void dfs1(int x,int pre){
int Max=0,cMax=0;
for (int i=0;i<G[x].size();i++){
edge e=G[x][i];
if (e.to==pre) continue;
dfs1(e.to,x);
if (e.w+dp[e.to][0]>=Max) {
cMax=Max;
Max=e.w+dp[e.to][0];
nxt[x]=e.to;
}
else if (e.w+dp[e.to][0]>cMax) cMax=e.w+dp[e.to][0];
}
dp[x][0]=Max;
dp[x][1]=cMax;
}
void dfs2(int x,int pre){
for (int i=0;i<G[x].size();i++){
edge e=G[x][i];
if (e.to==pre) continue;
if (nxt[x]==e.to){
dp[e.to][2]=max(dp[x][1]+e.w,dp[x][2]+e.w);
}
else dp[e.to][2]=max(dp[x][0]+e.w,dp[x][2]+e.w);
dfs2(e.to,x);
}
}
int main() {
while (~scanf ("%d",&n)){
memset(nxt,0,sizeof(nxt));
memset(dp,0,sizeof(dp));
for (int i=0;i<maxn;i++) G[i].clear();
for (int i=2;i<=n;i++){
int v,w;
scanf ("%d%d",&v,&w);
G[i].push_back(edge{v,w});
G[v].push_back(edge{i,w});
}
dfs1(1,1);
dfs2(1,1);
for (int i=1;i<=n;i++){
printf ("%d\n",max(dp[i][0],dp[i][2]));
}
}
return 0;
}