SICILY1138. 寻宝之旅 树形dp

题目描述

国家探险队长 Jack 意外弄到了一份秦始皇的藏宝图，于是，探险队一行人便踏上寻宝之旅，去寻找传说中的宝藏。

藏宝点分布在森林的各处，每个点有一个值，表示藏宝的价值。它们之间由一些小路相连，小路不会形成环，即两个藏宝点之间有且仅有一条道路。探险队从其中的一点出发，每次他们可以留一个人在此点开采宝藏，也可以不留，然后其余的人可以分成若干队向这一点相邻的点走去。需要注意的是，如果他们把队伍分成两队或两队以上，就必须留一个人在当前点，提供联络和通讯，当然这个人也可以一边开采此地的宝藏。并且，为了节约时间，队伍在前往开采宝藏过程中是不会走回头路的。现在你作为队长的助理，根据已有的藏宝图，请计算探险队所能开采的最大宝藏价值。

注意：在整个过程中，每个人最多只能开采一个点的宝藏。

输入格式

第 1 行有 2 个整数 n 和 m。其中 n 表示藏宝点的个数（1≤n≤100），m 表示探险队的人数（1≤m≤100）。

第 2 行是 n 个不超过 100 的整数，分别表示 1 到 n 每个点的宝藏价值。

接下来 n-1 行，每行两个数，x 和 y（1≤x，y≤n，x≠y），表示藏宝点 x 与 y 之间有一条路，数据保证不会有重复的路出现。

假设一开始探险队在点 1 处。

输出格式

输出一个整数，表示探险队所能获得最大宝藏价值。

样例数据 1

输入　 [复制]

5 3 
1 3 7 2 8 
1 2 
2 3 
1 4 
4 5

输出

16

备注

【数据范围】
对 40% 的输入数据 ：1≤n≤30；m≤12。

对 100% 的输入数据 ：1≤n≤100；m≤100。

    分析：此题一看在树上，求最优值，范围不大，一般就是树形dp了。然后我们考虑f[i][j]表示以i为根节点向树中派j个人去的最优值，不难看出当不留人时，f[i][j]=max(f[son[i][j]),现在考虑要留人的情况，首先只能下派j-1个人了，因为要有一人留守，然后类似于一个背包问题，相当于上限为j-1，有许多子树，从中取的最优值，然后做一遍背包就ok了，注意要维护确认背包中没有重复选中val[i]，否则会重复选，所以f[i][j][2]再开二维维护一下；

# include <iostream>
# include <cstdio>
# include <cstring>
# include <cmath>
# include <algorithm>
using namespace std;
int read()
{
	int i=0,f=1;char c=getchar();
	while(c<'0'||c>'9') 
	{
		if(c=='-') f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')
	{
		i=(i<<3)+(i<<1)+c-'0';
		c=getchar();
	}
	return i*f;
}
struct node
{
	int v,u;
	node *next;
}E[105<<2],*first[105],*cnt=E;
int f[105][105][2],res[105],val[105];
int n,m,x,y;
bool vis[105];
inline void addedge(int x,int y)
{
	node *p=++cnt;
	p->next=first[x];
	first[x]=p;p->v=y;
}
inline void DFS(int u)
{
	vis[u]=true;f[u][1][0]=val[u];
	for(node *p=first[u];p;p=p->next)
	{
		if(!vis[p->v])
		{
			DFS(p->v);
			for(int i=1;i<=m;++i) res[i]=max(f[p->v][i][0],f[u][i][0]);
			
			for(int i=1;i<=m;++i)
			    for(int j=1;j<i;++j)
			        res[i]=max(res[i],f[u][i-j-1][1]+f[p->v][j][0]+val[u]);
			
			for(int i=1;i<=m;++i) f[u][i][0]=res[i];
			
			for(int i=1;i<=m;++i) res[i]=max(f[p->v][i][0],f[u][i][1]);
			
			for(int i=1;i<=m;++i)
			    for(int j=1;j<i;++j)
			        res[i]=max(res[i],f[u][i-j][1]+f[p->v][j][0]);
			
			for(int i=1;i<=m;++i) f[u][i][1]=res[i];
		}
	}
}
int main()
{
	
	n=read(),m=read();
	for(int i=1;i<=n;++i) val[i]=read();
	for(int i=1;i<n;++i)
	{
		x=read(),y=read();
		addedge(x,y),addedge(y,x);
	}
	DFS(1);
	cout<<f[1][m][0]<<endl;
}