题意:给定一棵树,每个节点都赋予一个权值,求一个子树,使得子树所有节点的权值之和最大。题目中节点是按照整数坐标的形式给出的,相邻的坐标之间有边相连。
思路:树型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; }