hdu2196 Computer 【树形dp】

本文介绍了一种解决计算机网络中每台电脑到最远电脑距离的问题。通过两次深度优先搜索(DFS)算法,首先计算每个节点的子树中最远及次远距离,再调整这些距离来获得最终答案。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值