hdu2196 computer dfs 树形dp

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=10005;
struct E {int to,nxt,w;}edge[N+N];
int idx[N],tot,n;
bool vis[N];
int max1[N],max2[N],up[N];
void addedge(int from,int to,int w){
	edge[tot].to=to;edge[tot].w=w;edge[tot].nxt=idx[from];idx[from]=tot++;
}
void init(){
	tot=1;
	memset(idx,0,sizeof(idx));
	memset(edge,0,sizeof(E));
	memset(vis,0,sizeof(vis));
	memset(up,0,sizeof(up));
	memset(max1,0,sizeof(max1));
	memset(max2,0,sizeof(max2));
	for(int i=2;i<=n;i++){
		int y,w;
		scanf("%d%d",&y,&w);
		addedge(i,y,w);addedge(y,i,w);
	}
}
void dfs1(int x){
	vis[x]=1;
	for(int t=idx[x];t;t=edge[t].nxt){
		E e=edge[t];
		if(!vis[e.to]){
			dfs1(e.to);
			//if(x==1) printf("***%d\n",max1[e.to]+e.w);			
			if(max1[e.to]+e.w>max1[x]){
				max2[x]=max1[x];
				max1[x]=max1[e.to]+e.w;
			}
			else if(max1[e.to]+e.w>max2[x]) max2[x]=max1[e.to]+e.w;
			//if(x==1) printf("**%d %d\n",max1[x],max2[x]);
		}
	}
}
void dfs2(int x){
	vis[x]=1;
	for(int t=idx[x];t;t=edge[t].nxt){
		E e=edge[t];
		if(!vis[e.to]){
			if(e.to==5){
				x++;x--;
			}
			up[e.to]=up[x]+e.w;
			if(max1[x]!=max1[e.to]+e.w) up[e.to]=max(up[e.to],max1[x]+e.w);
			else up[e.to]=max(up[e.to],max2[x]+e.w);
			dfs2(e.to);
		}
	}
}
int main(){
	freopen("in.txt","r",stdin);
	freopen("out.txt","w",stdout);
	while(~scanf("%d",&n)){
		init(); 
		dfs1(1);
		memset(vis,0,sizeof(vis));
		dfs2(1);
		//printf("***%d\n",max2[1]);
		for(int i=1;i<=n;i++) printf("%d\n",max(up[i],max1[i]));
		//for(int i=1;i<=n;i++) printf("%d ",max2[i]);
	}
	return 0;
}

hdu2196computer

题意:求每个点到其他点的最长路径 n<=10000

题解:不想说它是树形dp 因为真心没怎么用dp思想

分类讨论吧 大概

以任意一点为根 转成有根树

由于对于一个节点来说,可能得到的距离最大的值的路径来自他的子树,或者从他的父节点过来,所以用两次DFS。

第一次DFS求出所有节点在他的子树范围内到叶子节点距离的最大值和第二大的值,第二次DFS更新从父节点过来的情况就可以了。

因为如果只存最大值的话,判断一个点的从父节点过来的最大值,那么如果他的父节点存的最大值正好是从该点过来的,那么就失去了从父节点过来的状态,所以要记录最大的两个值。


你可能感兴趣的:(dp,HDU,DFS)