ACM/ICPC 之 树形DP(POJ1192)

将某点看做根状态，邻接点看做子状态，由子状态向根状态转移。

 

---

 

POJ1192-最优连通子集

 

　　题解：将每一个点分成两个状态进行保存，因此可以构造一个数组dp[i][2]。

　　　　　dp[i][0]：不包括该点权值的子集最大权值和

　　　　　dp[i][1]：包括该点权值的子集最大权值和

 

//Time:16Ms Memory:184K
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;

#define MAX 1005
#define INF 0x3f3f3f3f

struct Node {
	int x, y;
	int w;
}node[MAX];

int n;
int dp[MAX][2];
bool v[MAX];

int DP(int num)
{
	v[num] = true;
	for (int i = 1; i <= n; i++)
	{
		if (!v[i] && abs(node[num].x - node[i].x) + abs(node[num].y - node[i].y) == 1)
		{
			dp[num][0] = DP(i);
			if(dp[i][1] > 0)	dp[num][1] += dp[i][1];
		}
	}
	return max(dp[num][0], dp[num][1]);
}

int main()
{
	scanf("%d", &n);
	memset(node, 0, sizeof(node));
	memset(v, false, sizeof(v));
	for (int i = 1; i <= n; i++)
		scanf("%d%d%d", &node[i].x, &node[i].y, &node[i].w);
	for (int i = 1; i <= n; i++)
	{
		dp[i][1] = node[i].w;
		dp[i][0] = -INF;	//题目后台数据只有一组，很弱，如果出现数据总和为负的情况下会有很多人WA掉
	}
	printf("%d\n", DP(1));
	return 0;
}