【题意】求树上每个点的最远距离是多少
这题一直被称为树形dp的经典是有它的道理的,因为树dp就是把dp放到树上做了,一般是从上到下或从下到上(利用回溯)的移转状态。而这题很合适的需要两次dfs。
对于<u,v>(有向),
dp[u][0]表示在u的子树下u的最远距离是多少
dp[u][1]表示在u的子树下(和dp[u][0]不是同一孩子)u的次远距离是多少
dp[u][2]表示通过u的父亲能走的最远距离是多少
第一次从下到上,对于<u,v>(有向),状态转移显然是 dp[u][0] = dp[v][0]+w[i];所以要先算出dp[v][0]才能知道dp[u][0]。故是从下往上。
第二次从上往下,其实就是再遍历一边图,把dp[v][2]算出来,显然:
dp[v][2] = max(dp[u][2],dp[v][0]+w[i]==dp[u][0]?dp[u][1]:dp[u][0]) + w[i];
要算dp[v][2],要先算dp[u][0],所以从上往下。
最后的答案就是 max(dp[u][0],dp[u][2])
【代码】
<span style="font-size:14px;">/* ***********************************************
Author :angon
************************************************ */
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)
#define lld %I64d
#define REP(i,k,n) for(int i=k;i<n;i++)
#define REPP(i,k,n) for(int i=k;i<=n;i++)
#define scan(d) scanf("%d",&d)
#define scanl(d) scanf("%I64d",&d)
#define scann(n,m) scanf("%d%d",&n,&m)
#define scannl(n,m) scanf("%I64d%I64d",&n,&m)
#define mst(a,k) memset(a,k,sizeof(a))
#define LL long long
#define N 10005
#define mod 1000000007
inline int read(){int s=0;char ch=getchar();for(; ch<'0'||ch>'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}
struct Edge
{
int v,w,next;
}edge[N*2];
int head[N],tot;
void addedge(int u,int v,int w)
{
edge[tot].v=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot++;
}
int dp[N][3];
void dfs1(int u,int v) //每个节点子树下的最大和次大
{
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
dfs1(v,u);
int w=edge[i].w;
int temp = dp[v][0] + w;
if(temp >= dp[u][0])
{
dp[u][1]=dp[u][0];
dp[u][0] = temp;
}
else if(temp > dp[u][1])
dp[u][1] = temp;
}
//printf("u=%d %d %d\n",u,dp[u][0],dp[u][1]);
}
void dfs2(int u,int v)
{
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(dp[u][0] == dp[v][0] + edge[i].w)
{
dp[v][2] = max(dp[u][2],dp[u][1]) + edge[i].w;
}
else
{
dp[v][2] = max(dp[u][2],dp[u][0]) + edge[i].w;
}
dfs2(v,u);
}
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int n;
while(~scan(n))
{
mst(head,-1); tot=0;
for(int v=2;v<=n;v++)
{
int u,w; scann(u,w);
addedge(u,v,w);
}
mst(dp,0);
dfs1(1,-1);
dfs2(1,-1);
for(int i=1;i<=n;i++)
printf("%d\n",max(dp[i][0],dp[i][2]));
}
return 0;
}</span>