【ACWing】886. 求组合数 II

题目地址：

https://www.acwing.com/problem/content/888/

给定$n$组询问，每次询问给定两个正整数$a$和$b$，求$\left(\genfrac{}{}{0ex}{}{a}{b}\right)$。答案模$10^9+7$返回。

输入格式：
第一行包含整数$n$。接下来$n$行，每行包含一组$a$和$b$。

输出格式：
共$n$行，每行输出一个询问的解。

数据范围：
$1\le n\le 10000$
$1\le b\le a\le 10^5$

直接用公式。我们的目标是求出$\left(\genfrac{}{}{0ex}{}{a}{b}\right)\mathrm{m}\mathrm{o}\mathrm{d}(10^9+7)$，而$10^9+7$是个素数。令$p=10^9+7$，那么我们要求$\frac{a!}{(a-b)!b!}\mathrm{m}\mathrm{o}\mathrm{d}p$。可以预处理出来$n!\mathrm{m}\mathrm{o}\mathrm{d}p$和$(n!{)}^{-1}\mathrm{m}\mathrm{o}\mathrm{d}p$，而预处理这个，可以通过递推关系来做。因为$n!=(n-1)!n$，而$(n!{)}^{-1}=((n-1)!{)}^{-1}(n{)}^{-1}$（这里的逆是指模$p$的逆），所以每次只需要算$n^{-1}$即可。关于模素数$p$的逆元求法，可以参考https://blog.csdn.net/qq_46105170/article/details/113823892。回答询问的时候，只需要输出$a!((a-b)!{)}^{-1}(b!{)}^{-1}\mathrm{m}\mathrm{o}\mathrm{d}p$。代码如下：

#include <iostream>
using namespace std;

const int N = 100010, mod = 1e9 + 7;

int fact[N], infact[N];

int fast_pow(int a, int k, int p) {
    int res = 1;
    while (k) {
        if (k & 1) res = (long) res * a % p;
        a = (long) a * a % p;
        k >>= 1;
    }

    return res;
}

int main() {
    fact[0] = infact[0] = 1;
    for (int i = 1; i < N; i++) {
        fact[i] = (long) fact[i - 1] * i % mod;
        infact[i] = (long) infact[i - 1] * fast_pow(i, mod - 2, mod) % mod;
    }

    int n;
    cin >> n;
    while (n--) {
        int a, b;
        cin >> a >> b;
        cout << ((long) fact[a] * infact[b] % mod * infact[a - b] % mod) << endl;
    }

    return 0;
}

预处理时间复杂度$O(N\log p)$，每次询问$O(1)$，空间$O(N)$，$N$是输入参数范围。