hdu2196 Computer 【樹形dp】

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


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