E. Karen and Supermarket
On the way home, Karen decided to stop by the supermarket to buy some groceries.
She needs to buy a lot of goods, but since she is a student her budget is still quite limited. In fact, she can only spend up to b dollars.
The supermarket sells n goods. The i-th good can be bought for ci dollars. Of course, each good can only be bought once.
Lately, the supermarket has been trying to increase its business. Karen, being a loyal customer, was given n coupons. If Karen purchases the i-th good, she can use the i-th coupon to decrease its price by di. Of course, a coupon cannot be used without buying the corresponding good.
There is, however, a constraint with the coupons. For all i ≥ 2, in order to use the i-th coupon, Karen must also use the xi-th coupon (which may mean using even more coupons to satisfy the requirement for that coupon).
Karen wants to know the following. What is the maximum number of goods she can buy, without exceeding her budget b?
The first line of input contains two integers n and b (1 ≤ n ≤ 5000, 1 ≤ b ≤ 109), the number of goods in the store and the amount of money Karen has, respectively.
The next n lines describe the items. Specifically:
- The i-th line among these starts with two integers, ci and di (1 ≤ di < ci ≤ 109), the price of the i-th good and the discount when using the coupon for the i-th good, respectively.
- If i ≥ 2, this is followed by another integer, xi (1 ≤ xi < i), denoting that the xi-th coupon must also be used before this coupon can be used.
Output a single integer on a line by itself, the number of different goods Karen can buy, without exceeding her budget.
6 16
10 9
10 5 1
12 2 1
20 18 3
10 2 3
2 1 5
4
5 10
3 1
3 1 1
3 1 2
3 1 3
3 1 4
5
In the first test case, Karen can purchase the following 4 items:
- Use the first coupon to buy the first item for 10 - 9 = 1 dollar.
- Use the third coupon to buy the third item for 12 - 2 = 10 dollars.
- Use the fourth coupon to buy the fourth item for 20 - 18 = 2 dollars.
- Buy the sixth item for 2 dollars.
The total cost of these goods is 15, which falls within her budget. Note, for example, that she cannot use the coupon on the sixth item, because then she should have also used the fifth coupon to buy the fifth item, which she did not do here.
In the second test case, Karen has enough money to use all the coupons and purchase everything.
题解:
一道比较经典的树形DP,一开始觉得就是做n次分组背包,但是发现是O(n^3)。
其实统计一下当前根节点所计算过的节点数,保存在size中就优化成O(n^2)了。
#include<iostream> #include<cstdio> #include<cstring> #include<cmath> #include<cstdlib> #include<algorithm> using namespace std; int n,m; struct node { int next,to; } edge[10001]; int head[10001],sze=1; void putin(int from,int to) { sze++; edge[sze].next=head[from]; edge[sze].to=to; head[from]=sze; } long long f[5001][5001][2],size[5001],a[10001],b[10001]; void dfs(int u) { int i; long long j,l; size[u]=1; f[u][0][0]=0; f[u][1][0]=a[u]; f[u][1][1]=a[u]-b[u]; for(i=head[u]; i!=-1; i=edge[i].next) { int y=edge[i].to; dfs(y); for(j=size[u]; j>=0; j--) { for(l=0; l<=size[y]; l++) { f[u][j+l][0]=min(f[u][j+l][0],f[u][j][0]+f[y][l][0]); f[u][j+l][1]=min(f[u][j+l][1],min(f[u][j][1]+f[y][l][0],f[u][j][1]+f[y][l][1])); } } size[u]+=size[y]; } } int main() { int i,j,fa; memset(head,-1,sizeof(head)); scanf("%d%d",&n,&m); scanf("%I64d%I64d",&a[1],&b[1]); for(i=2; i<=n; i++) { scanf("%I64d%I64d%d",&a[i],&b[i],&fa); putin(fa,i); } memset(f,127/3,sizeof(f)); dfs(1); int ans=0; for(i=0; i<=n; i++) { if(f[1][i][0]<=m||f[1][i][1]<=m)ans=i; } printf("%d\n",ans); return 0; }