Computer（树的直径做法）

Computer

 

 

 

 

重要性质：树上任意点能到的最远点，一定是树的直径的某个端点。 

解题思路：根据上述性质，先求出树的直径，然后从直径的两个端点u和v分别DFS整棵树，对于每个结点得到两个距离d[i].u和d[i].v， 二者的最大值即是i点能到的最远点的距离

AC_Code:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
using namespace std;
typedef long long ll;
const int maxn = 1e4+10;
const int inf = 0x3f3f3f3f;
const int mod = 998244353;
#define rep(i,first,second) for(int i=first;i<=second;i++)
#define dep(i,first,second) for(int i=first;i>=second;i--)

struct edge{ int to,nxt,val;}e[maxn<<1];
int head[maxn<<1],maxdis,maxv,tot;
int vis[maxn],d1[maxn],d2[maxn];
int n;

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

void dfs(int u,int f,int dis,int d[]){
    d[u]=dis;
    if( d[maxv]<d[u] ) maxv = u;
    for(int i=head[u];~i;i=e[i].nxt){
        int v=e[i].to,w=e[i].val;
        if( v==f ) continue;
        dfs(v,u,dis+w,d);
    }
}

int main()
{
    while( ~scanf("%d",&n) ){
        memset(head,-1,sizeof(head));
        tot=0;

        rep(i,2,n){
            int u,v,w;
            scanf("%d%d",&v,&w);
            u=i;
            add(u,v,w);
            add(v,u,w);
        }

        int s;
        s=1; maxv=0; d1[maxv]=0;
        dfs(s,-1,0,d1);

        s=maxv; d1[maxv]=0;
        dfs(s,-1,0,d1);

        s=maxv; d2[maxv]=0;
        dfs(s,-1,0,d2);

        rep(i,1,n){
            printf("%d\n",max(d1[i],d2[i]));
        }
    }
    return 0;
}