现在的位置: 首页 > 综合 > 正文

Codeforces Round#131D

2018年01月14日 ⁄ 综合 ⁄ 共 1581字 ⁄ 字号 评论关闭

题意:

给定数n和10个数的数组a,求满足以下条件的数的个数
1、数的长度不超过n
2、这个数不含前导0
3、数字i至少出现a[i]次

典型的数位DP,我们设f[i][j]为前i位数使用了j-9的数字能得到的数的个数
首先枚举长度len,对j的不同情况进行讨论:
1、j=9,此时若len>=a[9],那么f[len][9]=1否则为0
2、0<j<9,此时考虑枚举数字j时,j+1到9已经枚举完毕。此时j的个数可以为k(a[j]<=k<=len),
对于每个k放置在数列中的方法显然有C(len,k)种。那么状态转移方程即为f[len][j]+=f[len-k][j+1]*c[len][k]
其中c是组合数。
3、j=0,此时因为不能有前导0,所以对于0来说放置的方法只有len-1种,个数上限也是len-1,别的细节和第二种情况的是差不多的。
中间注意步步取模,注意是长度不超过n,所以最后输出sum(f[i][0]){0<=i<=n}即可

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<string>
#include<set>
#include<queue>
#include<stack>
#include<map>
#include<cmath>
#include<cstdlib>
#define ll long long
#define maxn 100010
#define inf 1000000000
#define linf (1LL<<50)
#define hzy 1000000007
using namespace std;
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x*=10;x+=ch-'0';ch=getchar();}
    return x*f;
}
 
inline void read(char *s,int &ts)
{
char x=getchar();
while(!(x>='a'&&x<='z'))x=getchar();
while(x>='a'&&x<='z')s[++ts]=x,x=getchar();
}
ll f[110][10];
ll c[110][110];
int n;
int a[20];
int main()
{
    scanf("%d",&n);
    for(int i=0;i<=9;i++)
    scanf("%d",&a[i]);
    for(int i=0;i<=n;i++)
    {
        c[i][0]=1;
        for(int j=1;j<=i;j++)
        c[i][j]=(c[i-1][j-1]+c[i-1][j])%hzy;
    }
    //for(int i=1;i<=n;i++)
    //for(int j=1;j<=i;j++)
    //printf("%d\n",c[i][j]);
    for(int len=0;len<=n;len++)
    {
        if(len>=a[9]) f[len][9]=1;
        else f[len][9]=0;
        for(int j=8;j>=1;j--)
        for(int k=a[j];k<=len;k++)
        f[len][j]=(f[len][j]+f[len-k][j+1]*c[len][k])%hzy;
        for(int k=a[0];k<=len-1;k++)
        f[len][0]=(f[len][0]+f[len-k][1]*c[len-1][k])%hzy;
    }
    ll ans=0;
    for(int i=0;i<=n;i++)
    ans=(ans+f[i][0])%hzy;
    printf("%I64d\n",ans);
    return 0;
}

抱歉!评论已关闭.