题目描述
传送门
题意:求树中每一个点到离它最远的点的距离。
题解
比较经典的树形dp。看过树的直径那道题dp做法的话应该是很好YY的。
maxn和_maxn分别表示点i到子树中的点距离的最大值和次大值,f和g分别表示点i到子树中的点距离的最大值和到不是子树中的点的距离的最大值。
递推的过程画个图YY一下。
代码
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int max_n=1e4+5;
const int max_e=max_n*2;
int n,x,y,z;
int tot,point[max_n],nxt[max_e],v[max_e],c[max_e];
int maxn[max_n],_maxn[max_n],f[max_n],g[max_n],ans[max_n];
inline void clear(){
tot=0;memset(point,0,sizeof(point));memset(nxt,0,sizeof(nxt));memset(v,0,sizeof(v));
for (int i=1;i<=n;++i) maxn[i]=_maxn[i]=f[i]=g[i]=ans[i]=0;
}
inline void add(int x,int y,int z){++tot;nxt[tot]=point[x];point[x]=tot;v[tot]=y;c[tot]=z;}
inline void treedp(int x,int fa){
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa){
treedp(v[i],x);
if (maxn[x]<maxn[v[i]]+c[i]){
_maxn[x]=maxn[x];
maxn[x]=maxn[v[i]]+c[i];
}
else _maxn[x]=max(_maxn[x],maxn[v[i]]+c[i]);
}
}
inline void get_ans(int x,int fa){
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa){
f[v[i]]=maxn[v[i]];
if (maxn[v[i]]+c[i]==maxn[x])
g[v[i]]=c[i]+max(g[x],_maxn[x]);
else
g[v[i]]=c[i]+ans[x];
ans[v[i]]=max(f[v[i]],g[v[i]]);
get_ans(v[i],x);
}
}
int main(){
while (scanf("%d",&n)!=EOF){
clear();
for (int i=2;i<=n;++i) scanf("%d%d",&x,&y),add(x,i,y),add(i,x,y);
treedp(1,0);
f[1]=ans[1]=maxn[1];
get_ans(1,0);
for (int i=1;i<=n;++i)
printf("%d\n",ans[i]);
}
}