码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces 300C Beautiful Numbers 【组合数】+【逆元】

时间:2018-10-09 19:58:49      阅读:135      评论:0      收藏:0      [点我收藏+]

标签:i++   ons   数组初始化   int   组合   cell   long   scan   lld   

<题目链接>

题目大意:

给出a和b,如果一个数每一位都是a或b,那么我们称这个数为good,在good的基础上,如果这个数的每一位之和也是good,那么这个数是excellent。求长度为n的excellent数的个数mod(1e9+7)。

解题分析:

我们可以枚举a的个数m,所以b的个数为(n-m),然后判断这种情况是否可行,即,是否满足a*m+b*(n-m)为good number ,如果满足的话,则答案加上C(n,m)。因为n很大,并且在计算组合数的过程中,需要除以很大的数,所以需要求逆元,因为本题模数为1e9+7,为质数,所以可以用费马小定理求逆元。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define LL long long
using namespace std;
const LL MOD=1e9+7;
const LL MAXN=1e6+10;
 
LL a,b;
LL n;
LL fact[MAXN];//存i的阶乘
 
bool Sum_ok(LL x){   
    while(x){    //判断每一位是否只由a或b组成
        LL res=x%10;
        if(res!=a&&res!=b) return false;
        x=x/10;
    }
    return true;
}
LL pow(LL m,LL n){
    if(n==0) return 1;
    LL ans=1;
    while(n>0){
        if(n&1) ans=ans*m%MOD;
        m=m*m%MOD;
        n=n/2;
    }
    return ans;
}
int main(){
    scanf("%d %d %d",&a,&b,&n);
    fact[0]=1;
    for(LL i=1;i<=n;i++) fact[i]=fact[i-1]*i%MOD;//将fact数组初始化将阶乘的值存进数组,不要忘了取余
    LL ans=0;
    for(LL i=0;i<=n;i++){    //枚举a的个数
        if(is_gnum(a*i+b*(n-i))){    //判断枚举出的数是不是good number
            LL t1=pow(fact[i],MOD-2)%MOD;       //费马小定理求逆元,注意求逆元的格式,fact[i]、fact[n-i]均为除数 
            LL t2=pow(fact[n-i],MOD-2)%MOD;
            ans += fact[n]*t1%MOD*t2%MOD;//利用上面讲解中推导出的公式计算出答案,利用组合数公式,n!/(m!*(n-m)!) 
        }
    }
    printf("%lld\n",ans%MOD);
    return 0;
}

 

 

2018-10-09

Codeforces 300C Beautiful Numbers 【组合数】+【逆元】

标签:i++   ons   数组初始化   int   组合   cell   long   scan   lld   

原文地址:https://www.cnblogs.com/00isok/p/9762172.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!