题意:
商店出售3种颜色的球,分别为红、绿、蓝。城市里有n个商店,第i个商店在第First_i天开始营业,连续营业ai+bi+ci天,每个商店每天只能出售一种颜色的球,其中ai天红,bi天绿,ci天蓝。每天最多有两个商店同时营业。如果同一天内有两个商店同时营业,那么这两个商店必须出售相同颜色的球。求不同的出售方案数(对1,000,000,007取模)。两种方案不同,当且仅当某一天某一个商店出售的球的颜色不同。
1≤n≤50
1≤First_i≤500
0≤ai,bi,ci≤100
First_i + ai+bi+ci - 1≤500
保证每天最多有两个商店同时营业。
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<iostream>
#include<algorithm>
#define N 110
#define mmod 1000000007
#define LL long long
using namespace std;
struct node{int st,ed,a,b,c;}A[N];
LL f[2][N][N],ans,C[3*N][3*N];
int n;
bool cmp(node x,node y)
{
if(x.st<y.st) return 1;
if(x.st>y.st) return 0;
if(x.ed<y.ed) return 1;
return 0;
}
void upd(LL &x,LL y)
{
x=(x+y)%mmod;
}
void dp()
{
int pre=0,now=1,p=A[1].st-1;
f[0][0][0]=1;
int cnt=0,i=1;
ans=1;
while(i<=n)
{
while(p<A[i].ed && (i==n || p<A[i+1].st-1))
{
for(int j=0;j<=100;j++)
for(int k=0;k<=100;k++)
f[now][j][k]=0;
for(int j=0;j<=A[i].a;j++)
for(int k=0;k<=A[i].b;k++)
{
if(f[pre][j][k]==0) continue;
if(j<A[i].a) upd(f[now][j+1][k],f[pre][j][k]);
if(k<A[i].b) upd(f[now][j][k+1],f[pre][j][k]);
if(cnt-j-k<A[i].c) upd(f[now][j][k],f[pre][j][k]);
}
p++;cnt++;
now^=1;pre^=1;
}
if(p==A[i].ed)
{
ans=ans*f[pre][A[i].a][A[i].b]%mmod;
for(int j=0;j<=100;j++)
for(int k=0;k<=100;k++)
f[pre][j][k]=0;
f[pre][0][0]=1;
p=A[i+1].st-1;cnt=0;
i++;
}
else if(A[i].ed<A[i+1].ed)
{
for(int j=0;j<=100;j++)
for(int k=0;k<=100;k++)
f[now][j][k]=0;
for(int j=0;j<=A[i].a;j++)
for(int k=0;k<=A[i].b;k++)
{
if(f[pre][j][k]==0) continue;
int a=A[i].a-j,b=A[i].b-k,c=A[i].c-(cnt-j-k),tot=a+b+c;
if(a>A[i+1].a || b>A[i+1].b || c>A[i+1].c) continue;
upd(f[now][a][b],f[pre][j][k]*C[tot][a]%mmod*C[tot-a][b]%mmod);
}
now^=1;pre^=1;
p=A[i].ed;cnt=A[i].ed-A[i+1].st+1;
i++;
}
else
{
for(int j=0;j<=100;j++)
for(int k=0;k<=100;k++)
f[now][j][k]=0;
for(int j=0;j<=A[i].a;j++)
for(int k=0;k<=A[i].b;k++)
{
if(f[pre][j][k]==0) continue;
int a=A[i].a-j,b=A[i].b-k,c=A[i].c-(cnt-j-k),tot=A[i+1].ed-A[i+1].st+1;
if(a<A[i+1].a || b<A[i+1].b || c<A[i+1].c) continue;
upd(f[now][j+A[i+1].a][k+A[i+1].b],f[pre][j][k]*C[tot][A[i+1].a]%mmod*C[tot-A[i+1].a][A[i+1].b]%mmod);
}
now^=1;pre^=1;
if(A[i].ed==A[i+1].ed)
{
ans=ans*f[pre][A[i].a][A[i].b]%mmod;
for(int j=0;j<=100;j++)
for(int k=0;k<=100;k++)
f[pre][j][k]=0;
f[pre][0][0]=1;
p=A[i+2].st-1;cnt=0;
i+=2;
}
else
{
p=A[i+1].ed;cnt+=A[i+1].ed-A[i+1].st+1;
swap(A[i],A[i+1]);
i++;
}
}
}
}
int main()
{
C[0][0]=1;
for(int i=1;i<3*N;i++)
{
C[i][0]=1;
for(int j=1;j<3*N;j++) C[i][j]=(C[i-1][j]+C[i-1][j-1])%mmod;
}
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&A[i].st);
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].ed=A[i].st+A[i].a+A[i].b+A[i].c-1;
sort(A+1,A+n+1,cmp);
dp();
printf("%lld\n",ans);
return 0;
}
题解:
同一时间只有两个商店,这个条件很奇怪。
显然相交的商店才会有影响,看看他的影响是什么。
如果我们用f[i][a][b]表示到第i天,当前线段用了a个红,b个绿有多少种方案。由于用的总数是确定的,所以蓝用了多少也是确定的。在相交之前的转移是简单的。
到了i后,当前线段就要和下面那一段相交了。但是注意确定a,b后,第i+1到j天上一段红绿蓝各用多少个都是确定的,而且下面那一段和他方案必须相同,所以就可以直接推出f[j][c][d],同时将当前线段设为下一条。
下面其他的情况处理方法类似地讨论即可。
1、
2、
3、