Codeforces 924E Wardrobe 动态规划

题意

有n个箱子，其中有一些箱子是重要的，每个箱子有一个高度，现在可以把箱子按照任意顺序堆叠在一起，问最多能有多少个重要箱子底部的高度位于区间$[l,r]$。
$n,\sum h_i\le10000$

分析

不难发现箱子的摆放顺序一定是一堆不重要的+一堆重要的+一堆不重要的。
我们把箱子反过来，也就是变成问最多有多少个箱子的顶部位于区间$[s-r,s-l]$，其中$s=\sum h_i$。
然后对于种类相同的箱子，显然高度较小的箱子位于上面会更优，因为这样会使得高度不变且两个断点更加接近。
所以把所有箱子排个序，然后做一次背包就好了。
时间复杂度$O(n^2)$。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

const int N=10005;

int n,l,r,f[N];
struct data{int x,y;}a[N];

bool cmp(data a,data b)
{
    return a.y<b.y||a.y==b.y&&a.x>b.x;
}

int main()
{
    scanf("%d%d%d",&n,&l,&r);
    int s=0;
    for (int i=1;i<=n;i++) scanf("%d",&a[i].x),s+=a[i].x;
    for (int i=1;i<=n;i++) scanf("%d",&a[i].y);
    std::sort(a+1,a+n+1,cmp);
    f[0]=1;
    for (int i=1;i<=n;i++)
        for (int j=s;j>=a[i].x;j--)
            if (f[j-a[i].x]) f[j]=std::max(f[j],f[j-a[i].x]+a[i].y*(j>=s-r&&j<=s-l));
    int ans=0;
    for (int i=0;i<=s;i++) ans=std::max(ans,f[i]);
    printf("%d",ans-1);
    return 0;
}