<span>HDU2196 Computer(树形dp)</span>

题意:

求树中每个点到所有叶子节点的距离的最大值是多少。

思路:

由于刚学树形dp,就参考了斌巨的博客

连接:http://www.cnblogs.com/kuangbin/archive/2012/08/28/2659915.html

代码:

/* ***********************************************
Author        :devil
Created Time  :2016/3/21 15:51:24
************************************************ */
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <stdlib.h>
using namespace std;
#define N 10010
int head[N],maxm[N],maxid[N],smaxm[N],smaxid[N],r;
struct wq
{
    int v,w,next;
} eg[N*2];
void init()
{
    memset(head,-1,sizeof(head));
    memset(maxm,0,sizeof(maxm));
    memset(smaxm,0,sizeof(smaxm));
    r=0;
}
void add(int u,int v,int w)
{
    eg[r].v=v;
    eg[r].w=w;
    eg[r].next=head[u];
    head[u]=r++;
    eg[r].v=u;
    eg[r].w=w;
    eg[r].next=head[v];
    head[v]=r++;
}
void dfs1(int u,int fa)
{
    for(int i=head[u]; i!=-1; i=eg[i].next)
    {
        int to=eg[i].v;
        if(to==fa) continue;
        dfs1(to,u);
        if(eg[i].w+maxm[to]>smaxm[u])
        {
            smaxm[u]=eg[i].w+maxm[to];
            smaxid[u]=to;
            if(smaxm[u]>maxm[u])
            {
                swap(smaxm[u],maxm[u]);
                swap(smaxid[u],maxid[u]);
            }
        }
    }
}
void dfs2(int u,int fa)
{
    for(int i=head[u]; i!=-1; i=eg[i].next)
    {
        int to=eg[i].v;
        if(to==fa) continue;
        if(to==maxid[u])
        {
            if(eg[i].w+smaxm[u]>smaxm[to])
            {
                smaxm[to]=eg[i].w+smaxm[u];
                smaxid[to]=u;
                if(smaxm[to]>maxm[to])
                {
                    swap(smaxm[to],maxm[to]);
                    swap(smaxid[to],maxid[to]);
                }
            }
        }
        else
        {
            if(eg[i].w+maxm[u]>smaxm[to])
            {
                smaxm[to]=eg[i].w+maxm[u];
                smaxid[to]=u;
                if(smaxm[to]>maxm[to])
                {
                    swap(smaxm[to],maxm[to]);
                    swap(smaxid[to],maxid[to]);
                }
            }
        }
        dfs2(to,u);
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    int n,x,y;
    while(~scanf("%d",&n))
    {
        init();
        for(int i=2; i<=n; i++)
        {
            scanf("%d%d",&x,&y);
            add(i,x,y);
        }
        dfs1(1,-1);
        dfs2(1,-1);
        for(int i=1;i<=n;i++)
            printf("%d\n",maxm[i]);
    }
    return 0;
}

 

全部评论

相关推荐

牛客263158796号:我领羊一面后十天不挂也不推进 今天问hr说等前序的第一批意向发完看情况再看是否推进
点赞 评论 收藏
分享
一个菜鸡罢了:哥们,感觉你的简历还是有点问题的,我提几点建议,看看能不能提供一点帮助 1. ”新余学院“别加粗,课程不清楚是否有必要写,感觉版面不如拿来写一下做过的事情,教育经历是你的弱势就尽量少写 2. “干部及社团经历”和“自我评价”删掉 3. 论文后面的“录用”和“小修”啥的都删掉,默认全录用,问了再说,反正小修毕业前肯定能发出来 4. 工作经验和研究成果没有体现你的个人贡献,着重包装一下个人贡献
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务