HDU - 2196 Computer(樹形dp)

題目鏈接: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;
}



 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章