分组

题目大意

给n个数分组,每一组的代价是最大值与最小值的差。
求有多少种分组方案,使得代价和不超过m。

DP

先想想一个朴素的dp。
将这些数从大到小排序,设Fi,j,k表示做完了第i个数,目前有j个未填入最小值的组,代价和为k。
第一种转移,第i+1个数单独作为一组:
*1->Fi+1,j,k
第二种转移,第i+1个数不作为最小值的填入一个组:
*j->Fi+1,j,k
第三种转移,新建一个以第i+1个数为最大值的组:
*1->Fi+1,j+1,k+a[i+1]
第四种转移,将第i+1个数作为最小值填入一个组:
*j->Fi+1,j-1,k-a[i+1]
这样做复杂度无法承受,因为第三维到了sigma{a}那么大。

优化

考虑差分,设si=ai-ai-1
那么aj-ai=sj+sj-1+sj-2+……+si+1
现在我们不妨从小到大排序,再来看看dp变成了什么样:
第一种转移,第i+1个数单独作为一组:
*1->Fi+1,j,k+s[i+1]*j
第二种转移,第i+1个数不作为最大值的填入一个组:
*j->Fi+1,j,k+s[i+1]*j
第三种转移,新建一个以第i+1个数为最小值的组:
*1->Fi+1,j+1,k+s[i+1]*j
第四种转移,将第i+1个数作为最大值填入一个组:
*j->Fi+1,j-1,k+s[i+1]*j
接下来第三维就是严格不下降的,所以至多为m。
复杂度n^2m

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=200+10,maxm=1000+10,mo=1000000007;
int f[2][maxn][maxm],a[maxn],s[maxn];
int i,j,k,l,t,n,m,ans;
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;
}
int main(){
    freopen("group.in","r",stdin);freopen("group.out","w",stdout);
    n=read();m=read();
    fo(i,1,n) a[i]=read();
    sort(a+1,a+n+1);
    fo(i,1,n) s[i]=a[i]-a[i-1];
    f[0][0][0]=1;
    fo(i,0,n-1){
        fo(j,0,n)
            fo(k,0,m)
                f[1-i%2][j][k]=0;
        t=1-i%2;
        l=i%2;
        fo(j,0,n)
            fo(k,0,m){
                if (k+s[i+1]*j>m) break;
                if (f[l][j][k]){
                    (f[t][j][k+s[i+1]*j]+=f[l][j][k])%=mo;
                    (f[t][j][k+s[i+1]*j]+=(ll)f[l][j][k]*j%mo)%=mo;
                    (f[t][j+1][k+s[i+1]*j]+=f[l][j][k])%=mo;
                    if (j) (f[t][j-1][k+s[i+1]*j]+=(ll)f[l][j][k]*j%mo)%=mo;
                }
            }
    }
    fo(k,0,m) (ans+=f[n%2][0][k])%=mo;
    printf("%d\n",ans);
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值