【树上背包】P2015 二叉苹果树

题目链接：P2015 二叉苹果树

题意：找到树上的一个包含根节点的大小为Q的连通块，使得树边权值和最大。

思路：

• 树上背包即可。

• 本题解用的是分组背包。一个子结点选取 $1,2....Q$ 视为一组内互斥的物品，在两组内枚举选取恰好装满父节点的物品。

代码：

//多叉：分组背包

#include<iostream>
#include<vector>
using namespace std;
typedef long long LL;

const int N=120;
int n,Q;
int dp[N][N];

vector<pair<int,int> >vec[N];

void dfs(int u,int pre)
{
	//这里用size==1来辨别是否为叶子节点更好一些
	int s[2],w[2],pos=0;
	for(int i=0;i<vec[u].size();i++)
		if(vec[u][i].first!=pre) s[pos]=vec[u][i].first,w[pos++]=vec[u][i].second;
		
	if(pos==0) return ;
	dfs(s[0],u);
	dfs(s[1],u);
	
	for(int k=1;k<=Q;k++)
		for(int i=0;i<=k;i++)
			if(i==0) dp[u][k]=max(dp[u][k],dp[s[1]][k-1]+w[1]);
			else if(i==k) dp[u][k]=max(dp[u][k],dp[s[0]][k-1]+w[0]);
			else dp[u][k]=max(dp[u][k],dp[s[0]][i-1]+dp[s[1]][k-i-1]+w[0]+w[1]);
}

int main()
{
	for(int i=0;i<N;i++) for(int j=1;j<N;j++) dp[i][j]=-1e9;
	//注意初始条件，无效条件赋值为无穷,表示无效状态
    //注意别爆int
	
	scanf("%d%d",&n,&Q);
	
	int a,b,c;
	
	for(int i=1;i<n;i++)
	{
		scanf("%d %d %d",&a,&b,&c);
		vec[a].push_back({b,c});
		vec[b].push_back({a,c});
	}
	
	dfs(1,-1);
	
	//for(int i=1;i<=n;i++) for(int k=1;k<=Q;k++) cout<<"dp["<<i<<"]["<<k<<"] is "<<dp[i][k]<<endl;
	
	printf("%d",dp[1][Q]);
	
	return 0;
}