SGU 149 Computer Network(树形DP)

又一道卡了半天的题啊，又一道被称作水题的树形DP，膜拜大神Orz。。。。

主要维护好三个量

down1[ k ] 为以k向下所能到达的最远距离

down2[ k ] 为以k向下所能到达的次远距离

up[ k ]为以k向上所能到达的最远距离

down1，down2都不难实现，说下up(wa在这里一次。。。)

up有以下几种可能：

1.如果其父节点的down1不过k，那么down1[ father ]+val[ k ]显然可以(father为k的父节点，val[ k ]为k-father边的权值)

2.如果其父节点的down1过k，则down2[ father ]+val[ k ]可能

3.还有一种开始被我忽略的可能，就是up[ fa[ father ] ]+val[ father ]+val[ k ]，因为这题建树实在太容易居然忘记了这根本就是个无向图，wa在test5上了。。。

三种情况取最大值就过了。。。

总的说，还是挺有意思一题。。。。

 

//SGU 149 Computer Network
//树形DP
//by night_watcher

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

#define N 10001

struct NODE{
    int to,next;
}node[N];

int val[N];
int head[N];
int son[N];
int fa[N];
int up[N];
int down1[N];
int down2[N];
int n,cnt;
bool visit[N];

void dfs1(int k){
    int i,j,to,v;
    for(i=head[k];i!=-1;i=node[i].next){
        to=node[i].to;
        v=val[to];
        dfs1(to);
        int temp=v+down1[to];
        if(temp>down1[k]){
            down2[k]=down1[k];
            down1[k]=temp;
            son[k]=to;
        }
        else if(temp==down1[k]){
            son[k]=0;
        }
        else if(temp>down2[k]){
            down2[k]=temp;
        }
    }
}

void upsearch(int k){
    visit[k]=1;
    int father,i,j,v;
    father=fa[k];
    if(!visit[father]){
        upsearch(father);
    }
    v=down1[father];
    if(son[father]==k){
        v=down2[father];
    }
    v=max(v,up[father]);
    up[k]=v+val[k];
}


int main(){
    int i,j,to,fn,v;
    memset(head,-1,sizeof(head));
    cin>>n;
    for(to=2;to<=n;to++){
        cin>>fn>>val[to];
        node[cnt].to=to;
        node[cnt].next=head[fn];
        head[fn]=cnt++;
        fa[to]=fn;
    }
    dfs1(1);
    for(i=2;i<=n;i++){
        if(!visit[i])
        upsearch(i);
    }
    for(i=1;i<=n;i++){
        cout<<max(up[i],down1[i])<<endl;
    }
    return 0;
}