这题当时做的时候不是很理解题意,就撸了一发卢卡斯定理,结果不对,后来理解了好久还是不理解那句话,
百度了题解,发现是第i中颜色的最后一个后面必须是i+1种颜色,然后剩下的就是组合问题,
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<set>
#include<string>
#define LL long long
using namespace std;
LL c[1005][1005];
const LL mod=1000000007;
int main()
{
c[0][0]=1;
for(int i=1;i<=1000;i++)
{
c[i][0]=1;
for(int j=1;j<=i;j++)
c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
}
int k;
while(cin>>k)
{
int a[1005];
for(int i=1;i<=k;i++)
cin>>a[i];
LL res=1;
int total=0;
for(int i=1;i<=k;i++)
{
res=res*c[total+a[i]-1][a[i]-1]%mod;
total+=a[i];
}
cout<<res<<endl;
}
return 0;
}