[CF815C] Karen and Supermarket

在回家的路上，凯伦决定到超市停下来买一些杂货。她需要买很多东西，但因为她是学生，所以她的预算仍然很有限。

事实上，她只花了b美元。

超市出售N种商品。第i件商品可以以ci美元的价格购买。当然，每件商品只能买一次。

最近，超市一直在努力促销。凯伦作为一个忠实的客户，收到了n张优惠券。

如果凯伦购买i次商品，她可以用i次优惠券降低di美元。当然，不买对应的商品，优惠券不能使用。

然而，对于优惠券有一个规则。对于所有i>=2，为了使用i张优惠券，凯伦必须也使用第xi张优惠券（这可能意味着使用更多优惠券来满足需求。）

凯伦想知道。她能在不超过预算B的情况下购买的最大商品数量是多少？
输入输出样例
输入样例#1：复制

6 16
10 9
10 5 1
12 2 1
20 18 3
10 2 3
2 1 5

输出样例#1：复制

输入样例#2：复制

5 10
3 1
3 1 1
3 1 2
3 1 3
3 1 4

输出样例#2：复制

5
数据范围：1<=n<=5000

Solution

今天考试考了这道题，我也是醉了，一开始就在想这题，一个小时后调出来了，自信AC，稳了。我还手输了几组数据都过了，详细请看我理解下的题目。后来才发现是i用了优惠,j才能用优惠，我认为是选了i，j就能优惠，关键是样例还过了，哎！还是太菜了。
每个结点只能有一个父节点，容易想到建树，然后树性DP，n=5000可以想到O(\(n^2\))的算法，我们定义\(f[i][j][2]\)代表i结点,选了j个结点，包括它自己，最后一维代表当前结点是否优惠。
容易想到转移方程为
to为x子节点，k代表除to子树外选了k个，j代表to子树选了j个
\[f[x][k+j][0]=min(f[x][k+j][0],f[to][j][0]+f[x][k][0]);\]
\[f[x][k+j][1]=min(f[x][k+j][1],f[to][j][0]+f[x][k][1]);\]
\[f[x][k+j][1]=min(f[x][k+j][1],f[to][j][1]+f[x][k][1]);\]

注意:数组不要赋值0x5f或0x7f，这样两个加起来就爆int了。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
struct node
{
    int to,next;
}a[1000100];
int w[101000],v[100100],len,last[101010],son[100010],tot;
int f[5100][5100][2];
void add(int a1,int a2)
{
    a[++len].to=a2;
    a[len].next=last[a1];
    last[a1]=len;
}
void dp(int x,int father)
{
    son[x]=1;
    for(int i=last[x];i;i=a[i].next)
    {
        int to=a[i].to;
        if(to==father)
        continue;
        dp(to,x);
        for(int k=son[x];k>=0;k--)
        for(int j=son[to];j>=0;j--)
        {
            f[x][k+j][0]=min(f[x][k+j][0],f[to][j][0]+f[x][k][0]);
            f[x][k+j][1]=min(f[x][k+j][1],min(f[to][j][1],f[to][j][0])+f[x][k][1]);
        }
        son[x]+=son[to];
    }
}
int main()
{
    //freopen("shopping.in","r",stdin);
    //freopen("shopping.out","w",stdout);
    memset(f,0x3f,sizeof(f));
    int n,s,x;
    cin>>n>>s;
    cin>>w[1]>>v[1];
    w[1]-=v[1];
    f[1][0][0]=0;f[1][1][1]=w[1];f[1][1][0]=w[1];
    for(int i=2;i<=n;i++)
    {
        scanf("%d%d%d",&w[i],&v[i],&x);
        add(x,i);add(i,x);
    }
    for(int i=2;i<=n;i++)
    f[i][0][0]=0,f[i][1][1]=w[i]-v[i],f[i][1][0]=w[i];
    dp(1,0);
    for(int i=n;i>=0;i--)
    {
        if(f[1][i][0]<=s||f[1][i][1]<=s)
        {cout<<i;return 0;}
    }
}