ZSTU 4119 Juice

树形DP+分组背包

题意： 给一个树，0节点具有无限能源，边权值表示边最大能源流通量，非0节点权值表示点亮该节点需要的能源（允许0），求最多能点亮多少个节点。

思路： 状态dp[i][j]表示以i为根的子树话费j能源能点亮的节点最大值。每个节点的背包容量上限为该节点到其父亲的那条边的边权

其实很水的一道DP。本人卡了一天多的原因，完全是因为背包没有掌握好，对0权值节点处理不当T_T

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

const int MAXN = 1010;
int head[MAXN];
struct E{
	int v,w,next;
}edge[MAXN];
int tot;
int dp[MAXN][110];
int n;
int w[MAXN];

int max(int a, int b) { return a>b?a:b; }
int min(int a, int b) { return a<b?a:b; }

void add(int a, int b, int w)
{
	edge[tot].next = head[a];
	edge[tot].v = b;
	edge[tot].w = w;
	head[a] = tot ++;
}

void dfs(int v, int weight)
{
	int i,j,p;

	// 考虑子节点
	for(p=head[v]; p!=-1; p=edge[p].next)
	{
		dfs(edge[p].v,edge[p].w);

		for(i=weight; i>=0; i--)
		{
			// 本题中允许0权重节点存在，所以背包第三层循环要从大到小，使0容量子节点最后一个计算
			for(j=min(edge[p].w,weight-i); j>=0; j--)
			{
				dp[v][i+j] = max(dp[v][i+j], dp[v][i]+dp[edge[p].v][j]);
			}
		}
	}

	// 考虑节点本身的放与不放
	for(i=weight; i>=w[v]; i--) dp[v][i] = max(dp[v][i], dp[v][i-w[v]]+1);
}

int main()
{
	int i,j;
	int a,b;
	while(~scanf("%d", &n))
	{
		tot = 0;
		memset(dp, 0, sizeof(dp));
		memset(head, -1, sizeof(head));
		for(i=1; i<=n; i++)
		{
			scanf("%d%d%d", &a, &w[i], &b);
			add(a,i,b);
		}

		int ans = 0;
		for(int p=head[0]; p!=-1; p=edge[p].next)
		{
			dfs(edge[p].v,edge[p].w);
			int maxn = 0;
			for(j=edge[p].w; j>=0; j--)
			{
				if(dp[edge[p].v][j] > maxn) maxn = dp[edge[p].v][j];
			}
			ans += maxn;
		}

		printf("%d\n", ans);
	}
	return 0;
}