HDU 3660 树形DP

原题地址

题意


Alice与Bob在一棵树的树根一同出游,两人从Bob开始轮换选择道路一直走到树叶,Alice会尽可能使走过的总长最小,而Bob相反。不过他们都不能让总长落到[l, r]之外


现在给出这棵树和l,r,要求我们判断最终走过的总长落到[l, r]内,如果能,则输出这个总长,否则输出Oh, my god!

思路


  • 由于两人都是最理性地考虑自己的选择,并且在某点由哪个人选择道路是已知的,所以可以使用DP

  • 在某个节点,从子节点的预期长度中选出使最终路径长度落在[l, r]内并且最大(小)的一个,加上对应的这条道路的长度,就是本节点到叶子节点的预期长度。

  • 在hdu上交的时候要注意优化,但是似乎这题数据太强了,即使通过也是和时限差了50ms,可以去做相同的uva-1484,数据要友好很多。

AC代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

int n;
long long l, r;
int fir[500005], nex[500005], vv[500005];
int ff[500005], ww[500005];

void dfs(int o, int d, int v)
{
	if (fir[o] == -1)
	{
		if (v > r || v < l)
		{
			ff[o] = -1;
			return;
		}
		ff[o] = 0;
		return;
	}
	ff[o] = d % 2 ? -1 : 99999999;
	for (register int i = fir[o]; i != -1; i = nex[i])
	{
		dfs(vv[i], d + 1, v + ww[i]);
		if (ff[vv[i]] == -1 || ff[vv[i]] == 99999999)
		{
			continue;
		}
		int acc = ff[vv[i]] + v + ww[i];
		if (acc > r || acc < l)
		{
			continue;
		}
		if (d % 2)
		{
			ff[o] = max(ff[o], ww[i] + ff[vv[i]]);
		}
		else
		{
			ff[o] = min(ff[o], ww[i] + ff[vv[i]]);
		}
	}
	return;
}

int main()
{
	while (scanf("%d%lld%lld", &n, &l, &r) != EOF)
	{
		memset(fir, -1, sizeof(int) * (n + 2));
		for (register int i = 1; i < n; ++i)
		{
			int u;
			scanf("%d%d%d", &u, &vv[i], &ww[i]);
			nex[i] = fir[u];
			fir[u] = i;
		}
		dfs(0, 1, 0);
		if (ff[0] != -1 && ff[0] != 99999999)
		{
			printf("%d\n", ff[0]);
		}
		else
		{
			printf("Oh, my god!\n");
		}
	}
	return 0;
}

posted @ 2021-01-25 20:34  _int_me  阅读(42)  评论(0编辑  收藏  举报