树形dp——树的最远距离 hdu2196

【题意】求树上每个点的最远距离是多少

对于<u,v>(有向),

dp[u][0]表示在u的子树下u的最远距离是多少

dp[u][1]表示在u的子树下(和dp[u][0]不是同一孩子)u的次远距离是多少

dp[u][2]表示通过u的父亲能走的最远距离是多少

第一次从下到上,对于<u,v>(有向),状态转移显然是 dp[u][0] = dp[v][0]+w[i];所以要先算出dp[v][0]才能知道dp[u][0]。故是从下往上。

第二次从上往下,其实就是再遍历一边图,把dp[v][2]算出来,显然:

dp[v][2] = max(dp[u][2],dp[v][0]+w[i]==dp[u][0]?dp[u][1]:dp[u][0]) + w[i];

对上述转移作一个简单解释:当dp[v][0]+w[i]==dp[u][0]时,说明这个点在u在u的子树下能走的最远距离包含了v点,那么dp[v][2]即v通过u的父亲能走的最远距离就是(u,v)这条边+max(dp[u][1],dp[u][2]). 如果不等,同理。

所以要算dp[v][2],要先算dp[u][0],所以从上往下。

最后的答案就是 max(dp[u][0],dp[u][2])

AC代码:

//树中某个节点到所有节点的最远距离
//author:zzy
#include <cstdio>  
#include <cmath>  
#include <cctype>  
#include <algorithm>  
#include <cstring>  
#include <utility>  
#include <string>  
#include <iostream>  
#include <map>  
#include <set>  
#include <vector>  
#include <queue>  
#include <stack>
const int MAXN=10200;
#define eps 1e-10
#define inf 0x3f3f3f3f
#define ll long long
#define CL(a,b) memset(a,b,sizeof(a))
#define MAXN 100010
using namespace std;
struct edge
{  
    int a;
    int w;//权值  
};
vector<edge> G[MAXN]; 
int tot;//总边数  
void add_edge(int u,int v,int w)//添加从u->v,权值为w的边  
{  
    struct edge E;
    E.a=u;E.w=w;  G[v].push_back(E);E.a=v;G[u].push_back(E);
}  
int dist[MAXN][3];//dist[i][0,1,2]分别为正向最大距离,正向次大距离,反向最大距离  
//int longest[MAXN];//下标是点,值是最远距离时对应点的位置
int dfs1(int u,int fa)//返回u的正向最大距离  
{  
    if(dist[u][0]>=0) return dist[u][0];  
    dist[u][0]=dist[u][1]=dist[u][2]=0;  
    for(int i=0;i<G[u].size();i++)  
    {  
        int v= G[u][i].a;  
        if(v==fa)continue;  
        if(dist[u][0]<dfs1(v,u)+G[u][i].w)  
        {  
            dist[u][1] = max(dist[u][1] , dist[u][0]);  
            dist[u][0]=dfs1(v,u)+G[u][i].w;  
        }  
        else if( dist[u][1]< dfs1(v,u)+G[u][i].w )
            dist[u][1] = max(dist[u][1] , dfs1(v,u)+G[u][i].w);  
    }  
    return dist[u][0];  
}  
void dfs2(int u,int fa)  
{  
    for(int i=0;i<G[u].size();i++)  
    {  
        int v = G[u][i].a;  
        if(v==fa)continue;  
        if(dist[u][0] == dist[v][0] + G[u][i].w) dist[v][2] = max(dist[u][2],dist[u][1])+G[u][i].w;  
        else dist[v][2] = max(dist[u][2],dist[u][0])+G[u][i].w;  
        dfs2(v,u);  
    }  
}  
int main()  
{  
    int n;  
    while(scanf("%d",&n)==1&&n)  
    {  
        tot=0;  
        CL(dist,-1);  
        for(int i=1; i<=n; i++)   G[i].clear();
        for(int i=2; i<=n; i++)  
        {  
            int v,w;  
            scanf("%d%d",&v,&w);  
            add_edge(i,v,w);
        }  
        dfs1(1,-1);  
        dfs2(1,-1);  
        for(int i=1;i<=n;i++)  
            printf("%d\n",max(dist[i][0],dist[i][2]));  
    }  
    return 0;  
}  

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值