Computer(求树上所有点的最远点距离)

题目链接：Computer

题意：求树上所有点的最远点距离。

思路：树上任意点的最远点一定是直径端点，枚举一下端点就好了。

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

typedef long long ll;

const int N = 10010;

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

int n;
int fin, mx;//fin代表的是每次跑完dfs之后的距离根的最远点，mx是距离最远点的距离
int dp[N];
int ans[N];
int head[N], tot;

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 dfs(int u, int pre)
{
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        int v = edge[i].to;
        int w = edge[i].w;
        if(v != pre)
        {
            dp[v] = dp[u] + w;
            ans[v] = max(ans[v], dp[v]);//下面两个端点都会跑，直接在这更新答案数组
            if(dp[v] > mx)
            {
                mx = dp[v];
                fin = v;
            }
            dfs(v, u);
        }
    }
}

int main()
{
    int v, w;
    while(~scanf("%d", &n))
    {
        tot = 0;
        mx = -1;
        for(int i = 0; i <= n; i++)
        {
            head[i] = -1;
            ans[i] = 0;
        }
        for(int i = 0; i < n - 1; i++)
        {
            scanf("%d %d", &v, &w);
            add(i + 2, v, w);
            add(v, i + 2, w);
        }
        dp[1] = 0;
        dfs(1, 0);//求第一个端点，跑完之后fin是第一个端点
        dp[fin] = 0;
        dfs(fin, 0);//求第二个端点，跑完之后fin是第二个端点
        dp[fin] = 0;
        dfs(fin, 0);
        for(int i = 1; i <= n; i++)
        {
            if(i == fin) printf("%d", mx);
            else printf("%d", ans[i]);
            printf("\n");
        }
    }
    return 0;
}