HDU2196 Computer （树形DP-换根）
dp[u][0]表示u向下走的最大距离；
dp[u][1]表示u向下走的次大距离；
dp[u][2]表示u向上走的最大距离；
最后的答案就是每个点的max（dp[u][0],dp[u][2]）;
求解次大距离并记录idx在求解向上最大距离中是有必要的，可以画图分析。
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 const int MAXN=10000+10; 6 int dp[MAXN][3]; 7 int idx[MAXN];//用来记录u往下的最大距离经过哪个子节点 8 struct Edge{ 9 int v,w,next; 10 }edge[MAXN<<1]; 11 int cnt,head[MAXN]; 12 13 void add(int u,int v,int w){ 14 edge[cnt].v=v; 15 edge[cnt].w=w; 16 edge[cnt].next=head[u]; 17 head[u]=cnt++; 18 } 19 20 void dfs1(int u,int fa){ 21 int mx1=0,mx2=0; 22 for(int i=head[u];~i;i=edge[i].next){ 23 int v=edge[i].v; 24 if(v==fa) continue; 25 dfs1(v,u); 26 int c=dp[v][0]+edge[i].w; 27 if(mx1<=c) mx2=mx1,mx1=c,idx[u]=v; 28 else if(mx2<c) mx2=c; 29 } 30 dp[u][0]=mx1;//向下走的最大值 31 dp[u][1]=mx2;//向下走的次大值 32 } 33 34 void dfs2(int u,int fa){//求u向上可以走的最大值 35 for(int i=head[u];~i;i=edge[i].next){ 36 int v=edge[i].v; 37 if(v==fa) continue; 38 if(idx[u]==v) 39 dp[v][2]=max(dp[u][1],dp[u][2])+edge[i].w; 40 else 41 dp[v][2]=max(dp[u][0],dp[u][2])+edge[i].w; 42 dfs2(v,u); 43 } 44 } 45 46 int main(){ 47 int n,a,b; 48 while(~scanf("%d",&n)){ 49 cnt=0; 50 memset(head,-1,sizeof(head)); 51 for(int i=2;i<=n;i++){ 52 scanf("%d%d",&a,&b); 53 add(i,a,b);add(a,i,b); 54 } 55 memset(dp,0,sizeof(dp)); 56 dfs1(1,1); 57 dfs2(1,1); 58 for(int i=1;i<=n;i++) 59 printf("%d\n",max(dp[i][0],dp[i][2])); 60 } 61 return 0; 62 }
