poj 1192 求树的最大权子树(树形dp)

题意:给定一棵树,每个节点都赋予一个权值,求一个子树,使得子树所有节点的权值之和最大。题目中节点是按照整数坐标的形式给出的,相邻的坐标之间有边相连。

思路:树型dp。dp[i]表示以i为根的子树具有的最大权值。dp的思路犹如一维最大子区间,即对于根节点i的所有子树j,如果j中节点的最大权大于0,则加入,否则舍弃。

#include <stdio.h>
#include <string.h>
#define max(a,b) ((a)>(b)?(a):(b))
#define N 1005
struct edge{
	int y,next,w;
}e[N*2];
struct point{
	int x,y;
}p[N];
int n,top,first[N],s[N],dp[N];
int adjacent(int i,int j){//判断p[i]和p[j]是否相邻
	return abs(p[i].x-p[j].x)+abs(p[i].y-p[j].y) == 1;
}
void add(int x,int y){
	e[top].y = y;
	e[top].next = first[x];
	first[x] = top++;
	e[top].y = x;
	e[top].next = first[y];
	first[y] = top++;
}
void dfs(int x,int fa){
	int i,y;
	dp[x] = s[x];//x为根的话它自己肯定得加入
	for(i = first[x];i!=-1;i=e[i].next){
		y = e[i].y;
		if(y != fa){
			dfs(y,x);
			if(dp[y] > 0)
				dp[x] += dp[y];
		}
	}
}
int main(){
	freopen("a.txt","r",stdin);
	while(scanf("%d",&n)!=EOF){
		int i,j,res;
		memset(first,-1,sizeof(first));
		memset(dp,0,sizeof(dp));
		top = 0;
		for(i = 0;i<n;i++)
			scanf("%d %d %d",&p[i].x,&p[i].y,&s[i]);
		for(i = 0;i<n-1;i++)
			for(j = i+1;j<n;j++)
				if(adjacent(i,j))
					add(i,j);
		dfs(0,-1);
		res = -0x3fffffff;
		for(i = 0;i<n;i++)//扫一遍找最大的
			res = max(res,dp[i]);
		printf("%d\n",res);	
	}
	return 0;
}


你可能感兴趣的:(poj 1192 求树的最大权子树(树形dp))