博主暴走中……神他妈的HDU,不能用fread读入的!!!我因为这个错WA了10+发……
这题的题意就是求每一个节点在树上和它距离最大的点的距离。
其实这题的思路还是挺清晰的,我们可以对于一棵有根树的一个非根非叶子节点进行讨论。
这个节点的最大距离有两种情况,一种是从以该节点为根的子树的最大距离扩展过来,另一种就是从父亲节点扩展过来。
引用一下另一位大佬的图片(看2号节点):
对于从父亲节点扩展过来的情况,我们还要分类讨论,若父亲节点的最大距离不是从当前节点扩展过来,那么当前节点的答案就是max(父亲节点的最大距离+父亲节点到当前节点的距离,当前节点的最大距离);若父亲节点的最大距离是从当前节点扩展过来,那么当前节点的答案就是max(父亲节点的第二大距离(不是从当前节点扩展来的)+父亲节点到当前节点的距离,当前节点的最大距离)。最后递归求解即可。
可能写的有点恶心,望读者能耐心读完吧。
附上AC代码:
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define N 10010
using namespace std;
struct side{
int to,w,nt;
}s[N<<1];
int n,num,h[N],y,w,f[N][3];
/*
inline char nc(){
static char ch[100010],*p1=ch,*p2=ch;
return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &a){
static char c=nc();int f=1;
for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
a*=f;return;
}
*/
inline void add(int x,int y,int w){
s[num]=(side){y,w,h[x]},h[x]=num++;
}
inline void so1(int x){
for (int i=h[x]; i!=-1; i=s[i].nt){
so1(s[i].to);
int tmp=f[s[i].to][0]+s[i].w;
if (tmp>=f[x][0]) f[x][1]=f[x][0],f[x][0]=tmp;
else if (tmp>f[x][1]) f[x][1]=tmp;
}
return;
}
inline void so2(int x){
for (int i=h[x]; i!=-1; i=s[i].nt){
if (f[x][0]==f[s[i].to][0]+s[i].w) f[s[i].to][2]=max(f[x][1],f[x][2])+s[i].w;
else f[s[i].to][2]=max(f[x][0],f[x][2])+s[i].w;
so2(s[i].to);
}
return;
}
int main(void){
while (~scanf("%d",&n)){
memset(h,-1,sizeof(h)),num=0;
for (int i=2; i<=n; ++i) scanf("%d%d",&y,&w),add(y,i,w);
memset(f,0,sizeof(f)),so1(1),so2(1);
for (int i=1; i<=n; ++i) printf("%d\n",max(f[i][0],f[i][2]));
}
return 0;
}