bzoj4583 购物

因为同一时刻最多有两个商店，我们可以设计出$O(n^5)$的dp，用$dp[i][j][k][l][m]$ 表示到时刻$i$，第一个商店还剩下$j$个红球、$k$个绿球，第二个商店还剩下$l$个红球、$m$个绿球的方案数。显然蓝球数量也是确定的。不难实现$O(1)$转移。
考虑优化。如果只有一个商店，这一段时间的方案数是可以直接算出来的。那么，如果有两个商店，一个结束得早，一个结束得晚，早结束的商店需要把自己的卖光，所以晚结束的那个必须跟他卖一样的。也就是说这一段我们也是可以$O(1)$算出来的。这样我们就可以跳过有两个商店同时存在的时间段，只在只有一个商店的时间段dp，复杂度变为了$O(n^3)$。
具体写法上可以按照商店开门的时间依次考虑，注意不要遗漏情况。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=100,p=1000000007;
void inc(int &x,int y)
{
    x+=y;
    x=x>=p?x-p:x;
}
int pow(int base,int k)
{
    int ret=1;
    for (;k;k>>=1,base=(LL)base*base%p)
        if (k&1) ret=(LL)ret*base%p;
    return ret;
}
struct shop
{
    int sta,end,a,b,c,len;
    bool operator < (const shop &s) const
    {
        return sta<s.sta;
    }
}a[maxn];
int dp[5*maxn+10][maxn+10][maxn+10],fac[5*maxn+10],inv[5*maxn+10],
n,m;
int get(int n,int x,int y)
{
    return (LL)fac[n]*inv[x]%p*inv[y]%p*inv[n-x-y]%p;
}
int main()
{
    int now,u,res;
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].sta);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].a);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].b);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].c);
    for (int i=1;i<=n;i++)
    {
        a[i].len=a[i].a+a[i].b+a[i].c;
        a[i].end=a[i].sta+a[i].len-1;
        m=max(m,a[i].end);
    }
    fac[0]=inv[0]=1;
    for (int i=1;i<=m;i++) fac[i]=(LL)fac[i-1]*i%p;
    inv[m]=pow(fac[m],p-2);
    for (int i=m-1;i>=1;i--) inv[i]=(LL)inv[i+1]*(i+1)%p;
    sort(a+1,a+n+1);
    dp[a[1].sta][a[1].a][a[1].b]=1;
    now=a[1].sta;
    for (int i=1;i<=n;i=u)
    {
        u=i+1;
        while (u<=n&&a[u].end<=a[i].end)
        {
            while (now<a[u].sta)
            {
                for (int j=0;j<=a[i].a;j++)
                    for (int k=max(0,a[i].end-now+1-j-a[i].c);k<=a[i].b&&k<=a[i].end-now+1-j;k++)
                    {
                        if (j) inc(dp[now+1][j-1][k],dp[now][j][k]);
                        if (k) inc(dp[now+1][j][k-1],dp[now][j][k]);
                        if (a[i].len+a[i].sta-now-j-k) inc(dp[now+1][j][k],dp[now][j][k]);
                    }
                now++;
            }
            for (int j=a[u].a;j<=a[i].a;j++)
                for (int k=max(a[u].b,a[i].end-now+1-j-a[i].c);k<=a[i].b&&k<=a[i].end-now+1-j-a[u].c;k++)
                    inc(dp[a[u].end+1][j-a[u].a][k-a[u].b],(LL)dp[now][j][k]*get(a[u].len,a[u].a,a[u].b)%p);
            now+=a[u].len;
            u++;
        }
        if (u<=n&&a[u].sta<=a[i].end)
        {
            while (now<a[u].sta)
            {
                for (int j=0;j<=a[i].a;j++)
                    for (int k=max(0,a[i].end-now+1-j-a[i].c);k<=a[i].b&&k<=a[i].end-now+1-j;k++)
                    {
                        if (j) inc(dp[now+1][j-1][k],dp[now][j][k]);
                        if (k) inc(dp[now+1][j][k-1],dp[now][j][k]);
                        if (a[i].len+a[i].sta-now-j-k) inc(dp[now+1][j][k],dp[now][j][k]);
                    }
                now++;
            }
            for (int j=0;j<=a[i].a&&j<=a[u].a;j++)
                for (int k=max(0,max(a[i].end-now+1-j-a[i].c,a[i].end-now+1-j-a[u].c));k<=a[i].b&&k<=a[u].b&&k<=a[i].end-now+1-j;k++)
                    inc(dp[a[i].end+1][a[u].a-j][a[u].b-k],(LL)dp[now][j][k]*get(a[i].end-now+1,j,k)%p);
            now=a[i].end+1;
        }
        else
        {
            res=0;
            for (int j=0;j<=a[i].a;j++)
                for (int k=max(0,a[i].end-now+1-j-a[i].c);k<=a[i].b&&k<=a[i].end-now+1-j;k++)
                    inc(res,(LL)dp[now][j][k]*get(a[i].end-now+1,j,k)%p);
            if (u<=n)
            {
                dp[a[u].sta][a[u].a][a[u].b]=res;
                now=a[u].sta;
            }
            else
            {
                printf("%d\n",res);
                return 0;
            }
        }
    }
}