hdu2196 Computers 树形DP

来源：互联网发布：免费手机小号软件编辑：程序博客网时间：2024/05/29 07:36

这题自己没想出来，，连续做了几天的树形DP结果反而思路被限制了，现在想来这题其实不难= =！主要是既要存 i 到叶子的最大值以及取得这个最大值的关联儿子节点，还要存取得次大的值这些信息。因为最后扩展到全部节点的时候可能出现 j 的父亲节点 i 恰好是 i 取得max的关联儿子节点，这样状态就无法转移了，于是我们就需要一个次大值的信息来解决这种问题。（最大与次大严格不重复）接下来就很简单了

#include<stdio.h>#include<iostream>#include<algorithm>#include<string.h>using namespace std;int head[10050];struct NODE{    int to,len,next;}edge[20000];int N;int imax[10050],imax2[10050];int id[10050],id2[10050];void dfs1(int x){    imax[x]=imax2[x]=0;    int i,to;    for(i=head[x];i!=-1;i=edge[i].next)    {        to=edge[i].to;        dfs1(to);        if(imax2[x] < imax[to] + edge[i].len)        {            imax2[x]=imax[to] + edge[i].len;            id2[x]=to;            if(imax[x] < imax2[x])            {                swap(imax[x],imax2[x]);                swap(id[x],id2[x]);            }        }    }}void dfs2(int x){    int i,to;    for(i=head[x];i!=-1;i=edge[i].next)    {        to=edge[i].to;        if(to == id[x])        {            if(edge[i].len + imax2[x] > imax2[to])            {                imax2[to]=edge[i].len+imax2[x];                id2[to]=x;                if(imax2[to] > imax[to])                {                    swap(imax2[to],imax[to]);                    swap(id[to],id2[to]);                }            }        }        else        {            if(edge[i].len + imax[x] > imax2[to])            {                imax2[to]=edge[i].len+imax[x];                id2[to]=x;                if(imax2[to] > imax[to])                {                    swap(imax2[to],imax[to]);                    swap(id[to],id2[to]);                }            }        }        dfs2(to);    }}int main(){    while(~scanf("%d",&N))    {        int i,j,len;        int c;        memset(head,-1,sizeof(head));        for(i=2,c=0;i<=N;i++)        {            scanf("%d %d",&j,&len);            edge[c].to=i,edge[c].len=len,edge[c].next=head[j],head[j]=c++;        }        dfs1(1);        dfs2(1);        for(i=1;i<=N;i++) printf("%d\n",imax[i]);    }}

0 0