poj 1192 树形DP 最大连通子集

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define MAX(x,y) ((x)>(y)?(x):(y))
using namespace std;
struct node
{
	int to,next;
}e[10000];
int head[1010];
int dp[10000][2];
int value[1010],x[1010],y[1010];
int cnt;
void add_edge(int from,int to)
{
	e[cnt].to=to;
	e[cnt].next=head[from];
	head[from]=cnt++;
}
void dfs(int u,int v)
{
	dp[u][0]=0;
	dp[u][1]=value[u];
	for(int i=head[u];i!=-1;i=e[i].next)
	{
		int to=e[i].to;
		if(to==v)
		continue;
	    dfs(to,u);
		dp[u][1]=MAX(dp[u][1],dp[u][1]+dp[to][1]);
		dp[u][0]=MAX(dp[u][0],dp[to][1]);	
	}
}
int main()
{
	int n;
	memset(head,-1,sizeof(head));
	cnt=0;
	scanf("%d",&n); 
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d%d",&x[i],&y[i],&value[i]);
		for(int j=1;j<i;j++)
		{
			if(abs(x[i]-x[j])+abs(y[i]-y[j])==1)
			{
				add_edge(i,j);
				add_edge(j,i);
			}
		}
	}
	dfs(1,-1);
	printf("%d\n",MAX(dp[1][0],dp[1][1]));
	return 0; 
}