[CF815C]Karen and Supermarket

最新推荐文章于 2022-09-15 19:01:54 发布

OneInDark

最新推荐文章于 2022-09-15 19:01:54 发布

阅读量145

点赞数

分类专栏： # 树 # 背包问题

本文链接：https://blog.csdn.net/qq_42101694/article/details/107709692

版权

树同时被 2 个专栏收录

98 篇文章 0 订阅

订阅专栏

背包问题

30 篇文章 0 订阅

订阅专栏

题目

传送门 to luogu

思路

没啥好说的，直接树形 $\tt dp$ 。定义状态 $f (x, i, 0 / 1)$ 表示 $x$ 子树内买 $i$ 件物品， $x$ 的父亲是否使用了优惠券。

怎么转移呢？可以这样理解： $f (x, i, 1)$ 表示 $x$ 必须使用优惠券， $f (x, i, 0)$ 则不使用。最后用 $f (x, i, 0)$ 更新 $f (x, i, 1)$ 来表示“有优惠券但是不用”的情况。

可复杂度是啥？树上背包，每对点在 $\tt lca$ 处产生 $1$ 的贡献，一共 $n^2$ 对点，所以复杂度是 $\mathcal O(n^2)$ 的，没什么问题。

代码

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(long long x){
	if(x < 0) putchar('-'), x = -x;
	if(x > 9) writeint(x/10);
	putchar((x%10)^48);
}
template < class T >
void getMax(T&a,T b){ if(a < b) a = b; }
template < class T >
void getMin(T&a,T b){ if(b < a) a = b; }

const int MaxN = 5005;
struct Edge{
	int to, nxt;
	Edge(int T=0,int N=0){
		to = T, nxt = N;
	}
} edge[MaxN<<1];
int head[MaxN], cntEdge, n;
void addEdge(int a,int b){
	edge[cntEdge] = Edge(b,head[a]);
	head[a] = cntEdge ++;
	// edge[cntEdge] = Edge(a,head[b]);
	// head[b] = cntEdge ++;
}

const int infty = (1<<30)-1;
int money; // 拥有的钱（用来剪枝）
int cost[MaxN], d[MaxN], siz[MaxN];
int dp[MaxN][MaxN][2]; // 0:没券 1:有券
long long tmp[MaxN][2]; // dp转移临时数组
void dfs(int x){
	/* 在此处先暂时认为1必须买 */
	dp[x][1][0] = cost[x]; // 没券用
	dp[x][1][1] = cost[x]-d[x];
	int now = 1; // 已经有的物品数量
	for(int i=head[x],y; ~i; i=edge[i].nxt){
		dfs(y = edge[i].to), now += siz[y];
		for(int r=0; r<2; ++r) // 买x否？
		for(int p=r; p<=now-siz[y]; ++p)
		for(int q=0; q<=siz[y]; ++q)
			getMin(tmp[p+q][r],dp[x][p][r]
				+0ll+dp[y][q][r]);
		for(int r=0; r<2; ++r)
		for(int j=r; j<=now; ++j){
			dp[x][j][r] = tmp[j][r];
			tmp[j][r] = infty;
		}
	}
	/* 此处进行最后的更新 */
	for(int i=1; i<=now; ++i)
		getMin(dp[x][i][1],dp[x][i][0]);
	siz[x] = now; // 别忘了哦！
}

int main(){
	n = readint(), money = readint();
	for(int i=1; i<=n; ++i)
		head[i] = -1;
	for(int r=0; r<2; ++r)
	for(int i=0; i<=n; ++i)
		tmp[i][r] = infty;
	for(int i=1; i<=n; ++i){
		cost[i] = readint();
		d[i] = readint();
		if(i >= 2)
			addEdge(readint(),i);
	}
	dfs(1);
	for(int i=n; ~i; --i)
		if(dp[1][i][1] <= money){
			printf("%d\n",i);
			return 0;
		}
	puts("3Q");
	return 0;
}