CodeForces 213 B.Numbers（组合数学+dp）

Description

问用至少$a[i]$个$i$构造一个长度不超过$n$且没有签到零的数字有多少种方案，结果模$10^9+7$

Input

第一行输入一整数$n$，之后输入十个整数$a[0],a[1],...,a[9](1\le n\le 100,0\le a[i]\le 100,i=0,1,...,9)$

Output

输出方案数，结果模$10^9+7$

Sample Input

1
0 0 0 0 0 0 0 0 0 1

Sample Output

1

Solution

$dp[i][j]$表示用$j,j+1,...,9$这些数字且每个数字的个数达到要求，构造出长度为$i$的数字的方案数，那么枚举$j$的个数$k$及选取这些$j$在数字中的位置有转移方程$dp[i][j]+=dp[i-k][j+1]\cdot C_{i-[j=0]}^k,a[j]\le k\le i$，答案即为$\sum\limits_{i=1}^ndp[n][0]$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=105;
#define mod 1000000007
int fact[maxn],inv[maxn];
void init(int n=100)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
}
int C(int n,int m)
{
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
int n,a[10],dp[maxn][11];
int main()
{
    init();
    while(~scanf("%d",&n))
    {
        for(int i=0;i<10;i++)scanf("%d",&a[i]);
        memset(dp,0,sizeof(dp));
        for(int i=a[9];i<=n;i++)dp[i][9]=1;
        for(int j=8;j>=0;j--)
            for(int i=a[j];i<=n;i++)
                for(int k=a[j];k<=i;k++)
                {
                    dp[i][j]+=(ll)dp[i-k][j+1]*C(i-(j==0),k)%mod;
                    if(dp[i][j]>=mod)dp[i][j]-=mod;
                }
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            ans+=dp[i][0];
            if(ans>=mod)ans-=mod;
        }
        printf("%d\n",ans);
    }
    return 0;
}