JZOJ5436. 【NOIP2017提高A组集训10.30】Group DP

题意：把n个数分成若干组，要求每组的极差和不超过k的方案数。
初看的时候很熟悉，应该做过（flag）但是不记得了。
一个很显然的想法是设f[i][j][k]表示做完前i-1个，分了j组，和为k，发现无法转移，因为没有记录每组的最大最小值。。
那么重新来，设f[i][j][k]表示做完前i-1个，还有j组没有分配最大值，极差和为k的方案数。
然后我们发现好像还是不大好转移最小值，于是我们可以用差分的思想，首先把所有数排序，然后发现每一组的价格差就是相邻两个差之和，也就是最大值与最小值之间所有相邻数字差之和。然后在dp的时候往没能确定的组里面加入a[i]-a[i-1]，那么就可以写出转移式子。设x=a[i]-a[i-1]，还要向没分好的j个组里面添加的差值tmp=x*j。
那么新的差值和就是tmp+k.
共有四种方案：
第一种，新开一组，a[i]为开头元素。
$f[i][j+1][tmp+k]+=f[i-1][j][k]$
第二种，新开的这个组只有a[i]一个元素。
$f[i][j][tmp+k]+=f[i-1][j][k]$
第三种，a[i]加入旧的一组，并且不是结束的那个元素。
$f[i][j][tmp+k]+=f[i-1][j][k]*j(j>0)$
第四种，a[i]加入旧的一组，并且这个组不再添加其他元素。
$f[i][j-1][tmp+k]+=f[i-1][j][k]*j(j>0)$
注意到*j,意义是加入任何一个未分好的组，所以j种方案。
然后空间不够，记得滚动啦。
原题：jzoj4830

#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int mo=1e9+7;
const int N=1e5+5;
ll f[5][205][1015];
int n,m,a[N];
int main()
{
    freopen("group.in","r",stdin);
    freopen("group.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,n)scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    int now=0;
    f[now][0][0]=1;
    fo(i,1,n)
    {
        now^=1;
        int x=a[i]-a[i-1];
        memset(f[now],0,sizeof(f[now]));
        fo(j,0,i-1)
        {
            int tmp=x*j;
            fo(k,0,m)
            {
                if(tmp+k>m)continue;
                int v=tmp+k;
                f[now][j+1][v]=(f[now][j+1][v]+f[1-now][j][k])%mo;
                f[now][j][v]=(f[now][j][v]+f[1-now][j][k])%mo;
                if (j)
                {
                    f[now][j][v]=(f[now][j][v]+f[1-now][j][k]*j)%mo;
                    f[now][j-1][v]=(f[now][j-1][v]+f[1-now][j][k]*j)%mo;
                }
            }
        }
    }
    ll ans=0;
    fo(i,0,m)
    ans=(ans+f[now][0][i])%mo;
    printf("%lld\n",ans); 
}