CodeForces 300 C.Beautiful Numbers（组合数学）

Description

一个数是好数当且仅当其各位数字均为$a$或$b$，一个数是完美数当且仅当其是好数且其各位数之和也是好数，问长度为$n$的好数有多少个

Input

三个整数$a,b,n(1\le a<b\le n,1\le n\le 10^6)$

Output

输出长度为$n$的完美数的个数，结果模$10^9+7$

Sample Input

1 3 3

Sample Output

1

Solution

枚举$a$的个数$i$，暴力判$i\cdot a+(n-i)\cdot b$是否为好数，如果是好数则方案数为$C_n^i$

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=1000005;
#define mod 1000000007
int fact[maxn],inv[maxn];
void init(int n=1e6)
{
    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;
}
bool check(int n,int a,int b)
{
    while(n)
    {
        if(n%10!=a&&n%10!=b)return 0;
        n/=10;
    }
    return 1;
}
int main()
{
    init();
    int a,b,n;
    while(~scanf("%d%d%d",&a,&b,&n))
    {
        int ans=0;
        for(int i=0;i<=n;i++)
            if(check(i*a+(n-i)*b,a,b))ans=(ans+C(n,i))%mod;
        printf("%d\n",ans);
    }
    return 0;
}