Tree(树形dp)

题目链接：Tree

题意：给你一棵树 ，让你求树的直径的长度和数量。

思路：

1. 如果单纯求树的直径的话， 可以用两次dfs解决， 但是这样只能求出直径，无法求出数目了， 要想连同数目一起求出来， 还得枚举所有的叶子， 这样肯定超时， 所以用树形dp， 在记录直径的同时记录数量。 记录直径的原理就是， 任取一点当根， 一直搜到底， 然后开始往上更新长度， 当碰到一个点有不止一条链的时候，开始更新他的长度即数量， 当出现新的最大长度， 数量就是两个点的数量之积， 如果长度和最大长度相同， 则加上他们二者数量的乘积。
2. 我们对于一个节点求出他下面所有链的长度以及每个子树长度最长的链有多少
3. 那么对于当前这个节点，经过这个节点的最长链一定是两个或一个链的合并。

参考博客：HDU 3534 Tree （树形dp求树的直径）

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;

typedef long long ll;

const int N = 100010;

struct node
{
    int to, next, w;
} edge[N * 2];

int n;
int head[N], tot;
int dis[N];//dis[i]表示i这个节点下的最长链的长度
int num[N];//num[i]表示i这个节点有多少链的长度是最长的
int ans1, ans2;

void add(int u, int v, int w)
{
    edge[tot].to = v;
    edge[tot].w = w;
    edge[tot].next = head[u];
    head[u] = tot++;
}

void init()
{
    ans1 = -1;
    tot = 0;
    for(int i = 0; i <= n; i++) head[i] = -1;
}

void dfs(int u, int pre)
{
    num[u] = 1;
    dis[u] = 0;
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        int v = edge[i].to;
        int w = edge[i].w;
        if(v != pre)
        {
            dfs(v, u);
            int tmp = dis[v] + w;
            if(tmp + dis[u] > ans1)
            {
                ans1 = tmp + dis[u];
                ans2 = num[u] * num[v];
            }
            else if(tmp + dis[u] == ans1) ans2 += num[u] * num[v];
            if(tmp > dis[u])
            {
                dis[u] = tmp;
                num[u] = num[v];
            }
            else if(tmp == dis[u]) num[u] += num[v];
        }
    }
}

int main()
{
    int u, v, w;
    while(~scanf("%d", &n))
    {
        init();
        for(int i = 1; i < n; i++)
        {
            scanf("%d %d %d", &u, &v, &w);
            add(u, v, w);
            add(v, u, w);
        }
        dfs(1, 0);
        printf("%d %d\n", ans1, ans2);
    }
    return 0;
}