题目
思路
没啥好说的,直接树形 d p \tt dp dp 。定义状态 f ( x , i , 0 / 1 ) f(x,i,0/1) f(x,i,0/1) 表示 x x x 子树内买 i i i 件物品, x x x 的父亲是否使用了优惠券。
怎么转移呢?可以这样理解: f ( x , i , 1 ) f(x,i,1) f(x,i,1) 表示 x x x 必须使用优惠券, f ( x , i , 0 ) f(x,i,0) f(x,i,0) 则不使用。最后用 f ( x , i , 0 ) f(x,i,0) f(x,i,0) 更新 f ( x , i , 1 ) f(x,i,1) f(x,i,1) 来表示“有优惠券但是不用”的情况。
可复杂度是啥?树上背包,每对点在 l c a \tt lca lca 处产生 1 1 1 的贡献,一共 n 2 n^2 n2 对点,所以复杂度是 O ( n 2 ) \mathcal O(n^2) O(n2) 的,没什么问题。
代码
#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;
}