题意
定义“好数字”是仅由a和b两个数码组成的数字(例如a=1、b=3时13、3、113111都算好数字,而123、322、0都不算),而“极好的数字”是各位之和为“好数字”的“好数字”(比如a=1、b=3时,111是“好数字”,且1+1+1=3为“好数字”,故111是“极好的数字”)
现在给你a、b和位数n,要求输出所有位数为n的“极好的数字”,因为答案可能很大,你只需要输出答案对10^{9}+7取模后的结果。
思路
由1113、1311、1131、3111都是“极好的数字”可以发现“好数字”是否是“极好的数字”与数码的位置无关,这一点从定义上“各位之和是’好数字‘的’好数字‘“可知,于是想到从这一点入手,直接开始枚举a的个数,判断 ka + (n-k)b 是否为”好数字“,若是则证明由k个a和n-k个b组成的数字都是”极好的数字“,然后用组合数来求出满足条件的数的个数,这些满足条件的数的个数之和即为答案。
大概思路已经得到,接下来就是实现的问题了:主要是如何快速得到组合数。用杨辉三角求 是
级别的,无法忍受,又想到组合数的通用公式:
发现如果我们可以提前知道1-100w的阶乘,就可以快速求解。但是组合数公式内含有减法,此时我们需要求出这些阶乘对应的逆元。
显然
故
因此可以先得到1000000!模1e9+7下的逆元(可用费马小定理或exgcd来求),然后直接倒序存入数组即可,需要的时候直接调用。
代码
#include <iostream>
#define modf 1000000007
using namespace std;
int a, b, n;
long long ans;
long long jc[1000005]; //存入n!就可以直接拿来用了!
long long ny[1000005]; //存入n!的逆元就可以直接拿来用了!
long long fm(long long a, long long p) {
long long u = 1;
while (p) {
if (p & 1) {
u = u * a % modf;
}
p >>= 1;
a = a * a % modf;
}
return u;
}
void init() {
jc[1] = 1;
for (int i = 2; i <= 1000000; ++i) {
jc[i] = jc[i - 1] * i % modf;
}
ny[1000000] = (fm(jc[1000000], modf - 2) % modf + modf) % modf;
for (int i = 999999; i >= 0; --i) {
ny[i] = ny[i + 1] * (i + 1) % modf; // (n-1)?!(?!是逆元(心虚)) = n?! * n
}
}
long long C(int n, int m) {
return jc[n] * ny[m] % modf * ny[n - m] % modf;
}
bool f(int x) {
while (x) {
if (x % 10 != a && x % 10 != b)
return false;
x /= 10;
}
return true;
}
int main() {
cin >> a >> b >> n;
init();
for (int i = 0; i <= n; ++i) {
int u = a * i + b * (n - i);
if (f(u)) {
ans = (ans + C(n, i)) % modf;
}
}
cout << ans;
return 0;
}