hdu2196 树形dp经典|树的直径

/*
两种做法
1.求出树直径v1,v2,那么有一个性质:任取一点u,树上到u距离最远的点必定是v1或v2
那么可以一次dfs求树v1
第二次求dis1[],求出所有点到v1的距离,同时求出v2
第三次求出dis2[],求出所有点到v2的距离
2.树形dp,dp[u][0|1]表示结点u向下的最大距离和向上的最大距离
dp[u][0]可以直接由子树求出
dp[u][1]应该从父节点推到子节点,如果v是u的大儿子,那么dp[v][1]=max(sec,dp[u][1])+e[i].w;
否则就是dp[v][1]=max(Max,dp[u][1])+e[i].w
*/
#include<bits/stdc++.h>
using namespace std;
#define maxn 10005
struct Edge{int to,nxt,w;}edge[maxn<<];
int head[maxn],tot,n;
long long dp[maxn][];
void init(){
memset(head,-,sizeof head);
tot=;
}
void addedge(int u,int v,int w){
edge[tot].to=v;edge[tot].w=w;edge[tot].nxt=head[u];head[u]=tot++;
}
void dfs0(int u,int pre){
dp[u][]=;
for(int i=head[u];i!=-;i=edge[i].nxt){
int v=edge[i].to;
if(v!=pre){
dfs0(v,u);
dp[u][]=max(dp[u][],dp[v][]+edge[i].w);
}
}
}
void dfs1(int u,int pre){
long long Max=,Sec=,v1,v2;//u的大儿子下标,二儿子下标
for(int i=head[u];i!=-;i=edge[i].nxt){//这个循环处理出u的大儿子
int v=edge[i].to;
if(v==pre)continue;
int tmp=edge[i].w+dp[v][];
if(tmp>Max){//找到了更大的儿子树
Sec=Max,Max=tmp;
v2=v1,v1=v;
}
else if(tmp==Max || tmp>Sec)
Sec=tmp,v2=v;
}
//printf("%d %d\n",u,Max);
if(u!=-){//和u的上面进行比较
long long tmp=dp[u][],v=-;
if(tmp>Max){//找到了更大的儿子树
Sec=Max,Max=tmp;
v2=v1,v1=v;
}
else if(tmp==Max || tmp>Sec)
Sec=tmp,v2=v;
} for(int i=head[u];i!=-;i=edge[i].nxt){//这个循环求dp进行递归
int v=edge[i].to;
if(v==pre)continue;
if(v==v1)dp[v][]=Sec+edge[i].w;
else dp[v][]=Max+edge[i].w;
dfs1(v,u);
}
//printf("%d %d\n",u,Max);
}
int main(){
while(cin>>n){
init();
for(int v=;v<=n;v++){
int u,w;
cin>>u>>w;
addedge(u,v,w);
addedge(v,u,w);
}
memset(dp,,sizeof dp);
dfs0(,);dfs1(,);
for(int i=;i<=n;i++)
printf("%lld\n",max(dp[i][],dp[i][]));
}
}
上一篇:char,wchar_t,char8_t,char16_t,char32_t


下一篇:2021-07-07