bzoj 4247: 挂饰 背包dp

题意

JOI君有N个装在手机上的挂饰，编号为1…N。 JOI君可以将其中的一些装在手机上。
JOI君的挂饰有一些与众不同——其中的一些挂饰附有可以挂其他挂件的挂钩。每个挂件要么直接挂在手机上，要么挂在其他挂件的挂钩上。直接挂在手机上的挂件最多有1个。
此外，每个挂件有一个安装时会获得的喜悦值，用一个整数来表示。如果JOI君很讨厌某个挂饰，那么这个挂饰的喜悦值就是一个负数。
JOI君想要最大化所有挂饰的喜悦值之和。注意不必要将所有的挂钩都挂上挂饰，而且一个都不挂也是可以的。
1<=N<=2000

分析

不难发现假设我们已经确定了要选的挂件，那么最优情况肯定是把挂件按照挂钩数从大到小来挂。那么只要把挂件按挂件数从大到小排序后背包即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=2005;
const int inf=2000000000;

int n,f[N][N];
pair<int,int> a[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void updata(int &x,int y)
{
    x=max(x,y);
}

int main()
{
    n=read();
    for (int i=1;i<=n;i++) a[i].first=read(),a[i].second=read();
    sort(a+1,a+n+1);
    for (int i=1;i<=n+1;i++)
        for (int j=0;j<=n;j++)
            f[i][j]=-inf;
    f[n+1][1]=0;
    int ans=0;
    for (int i=n+1;i>=1;i--)
        for (int j=0;j<=n;j++)
            if (f[i][j]>-inf)
            {
                ans=max(ans,f[i][j]);
                updata(f[i-1][j],f[i][j]);
                if (j) updata(f[i-1][min(j+a[i-1].first-1,n)],f[i][j]+a[i-1].second);
            }
    printf("%d",ans);
    return 0;
}