HDU-1011-Starship Troopers

HDU-1011-Starship Troopers

传送门

这一道题是一道树形dp。

题目大意：给了你一颗树，每个结点上有虫子和脑袋。你需要在有限个人数的情况下获取最多的脑袋。问最多可以获得多少脑袋。一个人可以消灭20只虫子。然后一个人不能走别人或者自己走过的路。

我们用dp[i][j]表示以i为结点派j个人可以获得的最多脑袋的个数。
我们假设1是根结点。从1出发。

要注意几个注意点：
1.当一个结点上没有虫子的时候，有脑袋。我们也需要派一个人取获得脑袋。
2.一个结点上小于20只虫子的时候也需要1个人去消灭。

所以我们判断某个结点x上面至少需要多少人：这个结点上面的虫子个数是bug[x]，上面的脑袋数量是b[x]；
我们需要派的人：cost = (bug[x] + 19) / 20；+19的原因就是上面说的第二个注意点（比如2个虫子也是需要1个人的）
然后做一个初始化：
i从cost~m（m是所拥有最大的人数）
dp[x][i] = b[x];（这个结点上面的脑袋数）

然后dfs一下有边的结点。维护一下最大值：
当前结点脑袋数+子树结点的脑袋数。和现在所存储的脑袋数取最大值

注意当前结点所需的人数底线是cost；子结点的dp已经在dfs底层处理好啦
循环从大往小
所以：
j从m到cost
k从0到j-cost（因为j - k(当前结点需要的最小人数)需要大于等于cost嘛）
dp[x][j] = max(dp[x][j], dp[x][j - k], dp[子结点][k]);

最后注意上面说的第一个要点。
如果dp[x][0]不等于0，就说明存在这个结点没有虫子但是有脑袋的情况。
所以需要维护一下dp[x][0]和dp[x][1]的最大值。

输出答案ans维护一下根结点（上面假设的）最大值就可以啦~
（因为你不知道用了多少个人嘛=-=）

代码部分：

#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a))
using namespace std;
const int N = 1e2 + 10;

int n, m;
int bug[N];
int b[N];
int dp[N][N];
int vis[N];
vector<int> v[N];

void init()
{
	mst(vis, 0);
	mst(dp, 0);
	for (int i = 0; i <= n; i++)
	{
		v[i].clear();
	}
}

void dfs(int x)
{
	vis[x] = 1;
	int siz = v[x].size();
	int cost = (bug[x] + 19) / 20;
	
	for (int i = cost; i <= m; i++)
	{
		dp[x][i] = b[x];
	}
	for (int i = 0; i < siz; i++)
	{
		int t = v[x][i];
		if (vis[t])
		{
			continue;
		}
		dfs(t);
		for (int j = m; j >= cost; j--)
		{
			for (int k = 0; j - k >= cost; k++)
			{
				dp[x][j] = max(dp[x][j], dp[x][j - k] + dp[t][k]);
			}
		}
	}
	if (dp[x][0] > 0)
	{
		dp[x][1] = max(dp[x][1], dp[x][0]);
		dp[x][0] = 0;
	}
}

int main()
{
	while (~scanf ("%d%d", &n, &m))
	{
		if (n == -1 && m == -1)
		{
			break;
		}
		for (int i = 1; i <= n; i++)
		{
			cin >> bug[i] >> b[i];
		}
		init();
		for (int i = 1; i < n; i++)
		{
			int x, y;
			scanf ("%d%d", &x, &y);
			v[x].push_back(y);
			v[y].push_back(x);
		}
		if (!m)
		{
			cout << "0\n";
			continue;
		}
		dfs(1);
		int ans = 0;
		for (int i = 1; i <= m; i++)
		{
			ans = max(ans, dp[1][i]);
		}
		cout << ans << endl;
	}
	return 0;
}