“访问”美术馆 （洛谷p1270）（树形DP）

题目地址
配套博客

先建立好树。
树形分组背包。优化掉一维。用dp（i，j）表示以i为根节点，分配 j 的时间所取得的最大值。
组内每一个物品的价值为 dp（j，y），体积为 y ，且 y < x
注意如果选择了儿子，那么一定会用2 * w[i]的时间来走这一段路（来回）。
最后当走到叶子节点。里面的东西都是一模一样，所以用完全背包处理（只有一类，所以只有一个循环）。

#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const double eps = 1e-5;
const int N = 1e3 + 10;
const int mod = 10007;

int dp[N][N];
int ne[N],e[N],h[N],w[N],idx=1;
int t,id=2,cnt[N];

void add(int a,int b,int c){
	w[idx]=c,e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void creat(int p){
	int a; cin>>a>>cnt[id];
	add(p,id,a);
	if(cnt[id]) return ;
	int pp = id;
	++id; creat(pp);
	++id; creat(pp);
}

void dfs(int u,int m){
	if(m <= 0) return ;
	for(int i=h[u];i;i=ne[i]){
		int j = e[i];
		dfs(j,m-2*w[i]);
		for(int x=m;x>=0;x--)
			for(int y=0;y<=x-2*w[i];y++)
				dp[u][x] = max(dp[u][x],dp[u][x-y-2*w[i]] + dp[j][y]);
	}
	for(int i=5;i<=5*cnt[u]&&i<=m;i++) dp[u][i] = dp[u][i-5] + 1;
}

signed main(){
	IOS
    cin>>t;
    creat(1);
    dfs(1,t);
    cout<<dp[1][t-1]<<endl;
	return 0;
}