UVALive 4015 Caves--树形dp

最新推荐文章于 2020-05-20 19:00:50 发布

LaoJiu_

最新推荐文章于 2020-05-20 19:00:50 发布

阅读量585

点赞数 1

分类专栏：【动态规划】--高级DP 文章标签： dp 树形dp UVALive ACM

本文链接：https://blog.csdn.net/LaoJiu_/article/details/52886161

版权

【动态规划】--高级DP 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

原题链接：http://vjudge.net/problem/UVALive-4015

题意：n个部落，编号0到n-1，n-1条路连接n个部落，连成一棵树，机器人从树根出发，问在最多走x米最多经过多少部落。

分析：

dp[ i ][ j ][ 0 ]表示从i部落出发，经过j个部落且不回到起点 i 的最小路径；

dp[ i ][ j ][ 1 ]表示从i部落出发，经过j个部落且回到起点 i 的最小路径。

#define _CRT_SECURE_NO_DEPRECATE 

#include<iostream>
#include<vector>
#include<cstring>
#include<queue>
#include<stack>
#include<algorithm>
#include<cmath>
#include<string>
#include<stdio.h>
#define INF 99999999
#define eps 0.0001
using namespace std;

int n, q;
int dp[505][505][2];
int son[505];
int cnt[505];
vector<int> vec[505];

void tree_dp(int u)
{
	for (int i = 1; i <= n; i++)
		dp[u][i][0] = dp[u][i][1] = INF;
	son[u] = 1;
	dp[u][1][0] = dp[u][1][1] = 0;
	for (int i = 0; i < vec[u].size(); i = i + 2)
	{
		int v = vec[u][i];
		int d = vec[u][i + 1];
		tree_dp(v);
		son[u] += son[v];
		for (int j = son[u]; j >= 2; j--)
			for (int k = 1; k <= min(j - 1, son[v]); k++)
			{
				dp[u][j][0] = min(dp[u][j][0], min(dp[u][j - k][0] + dp[v][k][1] + 2 * d, dp[u][j - k][1] + dp[v][k][0] + d));
				dp[u][j][1] = min(dp[u][j][1], dp[u][j - k][1] + dp[v][k][1] + 2 * d);
			}
	}
}

int main() 
{
	int T = 1;
	while (~scanf("%d", &n) && n)
	{
		memset(cnt, 0, sizeof(cnt));
		memset(son, 0, sizeof(son));
		for (int i = 0; i < n; i++)
			vec[i].clear();

		int x, y, z;
		for (int i = 1; i < n; i++)
		{
			scanf("%d%d%d", &x, &y, &z);
			vec[y].push_back(x);
			vec[y].push_back(z);
			cnt[x]++;
		}
		
		int root = 0;
		for (int i = 0; i < n;i++)
			if (cnt[i] == 0)
			{
				root = i;
				break;
			}
		tree_dp(root);

		scanf("%d", &q);
		printf("Case %d:\n", T++);
		while (q--)
		{
			scanf("%d", &x);
			int ans = 1;
			for (int i = 1; i <= n; i++)
				if (dp[root][i][0] <= x)
					ans = i;
			printf("%d\n", ans);
		}
	}
	return 0;
}