[bzoj4583]购物/SRM601 WinterAndShopping

题目描述

商店出售3种颜色的球，分别为红、绿、蓝。城市里有n个商店，第i个商店在第First_i天开始营业，连续营业Red_i+Green_i+Blue_i天，每个商店每天只能出售一种颜色的球。每天最多有两个商店同时营业。如果同一天内有两个商店同时营业，那么这两个商店必须出售相同颜色的球。求不同的出售方案数（对1,000,000,007取模）。两种方案不同，当且仅当某一天某一个商店出售的球的颜色不同。

1≤n≤50
1≤First_i≤500
0≤Red_i, Green_i, Blue_i≤100
0 < Red_i + Green_i + Blue_i
First_i + Red_i + Green_i + Blue_i - 1≤500

分析

如果一天只有一个商店开，很容易可以设出状态f[i][r][g]，表示第i天，这天开的商店卖了r天红球，g天绿球，蓝球的天数可以通过r,g算出来。然后分三种情况考虑即可。

现在关键在于那些有两个商店的时刻。这些时刻可以看成一块，然后枚举r,g，乘上两个组合数的乘积。注意这里也有三种情况（两个商店先开的先关、后开的先关，同一天关）。由于n不大，这些块也不会很多。

注意细节。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N=55,M=105,mo=1e9+7;

typedef long long LL;

int n,f[M*5][M][M],C[M*5][M*5],ans,m,la,X[M*5],Y[M*5];

struct data
{
    int f,r,g,b,t,l;
}a[N];

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

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].f);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].r);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].g);
    for (int i=1;i<=n;i++) scanf("%d",&a[i].b);
    for (int i=1;i<=n;i++) a[i].t=a[i].f+a[i].r+a[i].g+a[i].b-1,a[i].l=a[i].t-a[i].f+1;
    C[0][0]=1;
    for (int i=1;i<=500;i++)
    {
        C[i][0]=C[i][i]=1;
        for (int j=1;j<i;j++) C[i][j]=(C[i-1][j-1]+C[i-1][j])%mo;
    }
    sort(a+1,a+n+1,cmp);
    for (int i=1;i<=n;i++)
        for (int j=a[i].f;j<=a[i].t;j++) if (X[j]==0) X[j]=i;else Y[j]=i;
    f[0][0][0]=1;
    for (int i=1,j;i<=501;i=j+1)
    {
        if (!X[i])
        {
            f[i][0][0]=f[i-1][0][0]; j=i;
            continue;
        }
        int x=X[i],R,G,B;
        if (!Y[i])
        {
            for (R=0;R<=a[x].r;R++)
            {
                for (G=0;G<=a[x].g;G++) if (f[i-1][R][G]>0)
                {
                    B=i-a[x].f-R-G;
                    if (B<0 || B>a[x].b) continue;
                    if (R<a[x].r) (f[i][R+1][G]+=f[i-1][R][G])%=mo;
                    if (G<a[x].g) (f[i][R][G+1]+=f[i-1][R][G])%=mo;
                    if (B<a[x].b) (f[i][R][G]+=f[i-1][R][G])%=mo;
                }
            }
            if (a[x].t==i)
            {
                int t=f[i][a[x].r][a[x].g];
                memset(f[i],0,sizeof(f[i]));
                f[i][0][0]=t;
            }
            j=i;
            continue;
        }
        int y=Y[i],tmp=(LL)C[a[y].l][a[y].b]*C[a[y].l-a[y].b][a[y].g]%mo;
        if (a[y].t==a[x].t)
        {
            j=a[y].t;
            R=a[x].r-a[y].r; G=a[x].g-a[y].g; B=a[x].b-a[y].b;
            if (R>=0 && B>=0 && G>=0 && B==i-a[x].f-R-G)
            {
                f[j][0][0]=(LL)f[i-1][R][G]*tmp%mo;
            }
            continue;
        }
        if (a[y].t<a[x].t)
        {
            j=a[y].t;
            for (R=0;R<=a[x].r-a[y].r;R++)
            {
                for (G=0;G<=a[x].g-a[y].g;G++) if (f[i-1][R][G]>0)
                {
                    B=i-a[x].f-R-G;
                    if (B+a[y].b>a[x].b || B<0) continue;
                    (f[j][R+a[y].r][G+a[y].g]+=(LL)f[i-1][R][G]*tmp%mo)%=mo;
                }
            }
            continue;
        }
        j=a[x].t;
        for (R=0;R<=a[x].r;R++)
        {
            if (a[x].r-R>a[y].r) continue;
            for (G=0;G<=a[x].g;G++) if (f[i-1][R][G]>0)
            {
                B=i-a[x].f-R-G;
                if (a[x].g-G>a[y].g || a[x].b-B>a[y].b || B<0 || B>a[x].b) continue;
                (f[j][a[x].r-R][a[x].g-G]+=(LL)f[i-1][R][G]*C[j-i+1][a[x].b-B]%mo*C[j-i+1-a[x].b+B][a[x].g-G]%mo)%=mo;
            }
        }
    }
    printf("%d\n",f[501][0][0]);
    return 0;
}