HDU 2196 Computer(树的最长路径/树的直径)

刚学树形dp的时候人家给这道题的定义是--简单题,当时心态就崩了,学完紫书上的另外两个分支之后回过头来看,发现也不是无从下手,思路问题,跑两种dfs,这个网络上有,思路写的十分清晰,但是代码看不明白,所以就跑去找kuangbin的模板,写的真好啊,一看就明白了!

思路:记录每一个树到子树的最长路径和次长路径,其中次长路径是通过和最长路径的不断更替得到的(这里kuangbin写的真的好,学到了),然后记录其父亲反向行走并且不经过最长子路径经过的节点的最长路径,哇,很容易理解其实,最后和最长路径进行比较得到最优解,还是要%kuangbin,奉为圭臬

#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define maxnn 10005
int maxn[maxnn];
int smaxn[maxnn];
int maxid[maxnn];
int smaxid[maxnn];
int head[maxnn];
int cnt;
struct node
{
    int to;
    int next;
    int w;
}e[maxnn * 2];
void addedge(int u, int v, int w)
{
    e[cnt].to = v;
    e[cnt].w = w;
    e[cnt].next = head[u];
    head[u] = cnt ++;
}
void init()
{
    cnt = 0;
    memset(head, -1, sizeof(head));
}
void dfs1(int u, int p)
{
    maxn[u] = smaxn[u] = 0;
    for(int i = head[u]; ~ i; i = e[i].next)
    {
        int v = e[i].to;
        if(v == p) continue;
        dfs1(v, u);
        if(smaxn[u] < maxn[v] + e[i].w)
        {
            smaxn[u] = maxn[v] + e[i].w;
            smaxid[u] = v;
            if(smaxn[u] > maxn[u])
            {
                swap(smaxn[u], maxn[u]);
                swap(smaxid[u], maxid[u]);
            }
        }
    }
}
void dfs2(int u, int p)
{
    for(int i = head[u]; ~ i; i = e[i].next)
    {
        int v = e[i].to;
        if(v == p) continue;
        if(v == maxid[u])
        {
            if(e[i].w + smaxn[u] > smaxn[v])
            {
                smaxn[v] = e[i].w + smaxn[u];
                smaxid[v] = u;
                if(smaxn[v] > maxn[v])
                {
                    swap(smaxn[v], maxn[v]);
                    swap(smaxid[v], maxid[v]);
                }
            }
        }
        else
        {
            if(e[i].w + maxn[u] > smaxn[v])
            {
                smaxn[v] = e[i].w + maxn[u];
                smaxid[v] = u;
                if(smaxn[v] > maxn[v])
                {
                    swap(smaxn[v], maxn[v]);
                    swap(smaxid[v], maxid[v]);
                }
            }
        }
        dfs2(v, u);
    }
}
int main()
{
    int n;
    while(scanf("%d",&n) != EOF)
    {
        init();
        int v, len;
        for(int i = 2; i <= n; i ++)
        {
            scanf("%d%d",&v,&len);
            addedge(i, v, len);
            addedge(v, i, len);
        }
        dfs1(1, -1);
        dfs2(1, -1);
        for(int i = 1; i <= n; i ++)
            printf("%d\n",maxn[i]);
    }
    return 0;
}

 

你可能感兴趣的:(树形dp,紫书阅读笔记)