寒训4C Beautiful Numbers

题意

定义“好数字”是仅由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;
}