不同范围组合数

一.递推法

题目描述

给定n组询问，每组询问给定两个整数a，b，请你输出$C_a^b$ mod (10^9+7)的值。
1≤n≤10000,
1≤b≤a≤2000

问题解法

根据公式$C_a^b$ = $C_{a-1}^b$ + $C_{a-1}^b$递推进行预处理，求出所有$C_a^b$ 的情况，询问的时间复杂度为O（1）。

时间复杂度

时间取决于a的值，两层for循环的时间复杂度是O（a^2)

参考代码

#include <iostream>
#include <cstring>

using namespace std;
const int N = 2030;
const int MOD = 1e9 + 7;
int f[N][N], n;

void init()
{
    for (int i = 0; i < N; i ++)
    {
        for (int j = 0; j <= i; j ++)
        {
            if (!j) f[i][j] = 1;
            else f[i][j] = (f[i - 1][j - 1] + f[i - 1][j]) % MOD;
        }
    }
}

int main()
{
    init();
    cin >> n;
    while (n --)
    {
        int a, b;
        cin >> a >> b;
        cout << f[a][b] << endl;
    }
}

---

---

二.预处理阶乘

题目描述

给定n组询问，每组询问给定两个整数a，b，请你输出$C_a^b$ mod (109+7)的值。
1≤n≤10000,
1≤b≤a≤10^5

问题解法

此时如果根据公式$C_a^b$ = $C_{a-1}^b$ + $C_{a-1}^b$来进行递推，显然是会TLE的。观察到
$C_a^b$ = $a!$ / ($(a-b)!$ * $(b)!$)
可以预处理求n的阶乘和n的阶乘的逆元（根据费马小定理，若p为质数，$b^{-1}$ 同余 $b^{p-2}$ （modp）求出$C_a^b$

时间复杂度

预处理时每次都要求阶乘的逆元时，都要进行进行一次快速幂的运算，循环a次，因此总的时间复杂度为O(a *$\log$MOD)。

参考代码

#include <iostream>
#include <cstring>

using namespace std;
int n;
const int N = 1e5 + 10;
int fact[N]; //n的阶乘
int infact[N]; //n的阶乘的逆元
int MOD = 1e9 + 7;
typedef long long LL;

int qumi(int p, int k, int m) //快速幂
{
    int res = 1;
    while (k)
    {
        if (k & 1)  res = (LL)res * p % m;
        p = (LL)p * p % m;
        k >>= 1;
    }
    return res;
}

int main()
{
    fact[0] = infact[0] = 1; 
    cin >> n;
    for (int i = 1; i < N; i ++)
    {
        fact[i] = (LL)fact[i - 1] * i % MOD;
        infact[i] = (LL)infact[i - 1] * qumi(i, MOD - 2, MOD) % MOD;  // (a * b)^-1 和 (a ^ -1) * (b ^ -1) 同余 
    }
    
    while (n --)
    {
        int a, b;
        cin >> a >> b;
        cout << (LL)fact[a] * infact[b] % MOD * infact[a - b] % MOD <<endl;
    }
}

---

---

三.利用Lucas定理

题目描述

给定n组询问，每组询问给定三个整数a,b,p，其中p是质数，请你输出$C_a^b$ mod p的值。
1≤n≤20,
1≤b≤a≤10^18,
1≤p≤10^5,

问题解法

此时a很大，不能再用上述方法，需要用到卢卡斯定理
百科上定义

即$C_a^b$ 同余于 $C_{a/p}^{b/p}$ * $C_{a(modp)}^{b(modp)}$ （mod p）
卢卡斯定理证明略

问题然后递归求解即可

参考代码

#include <iostream>
#include <cstring>

using namespace std;
typedef long long LL;

LL a, b;
int p, n;
int quim(int a, int k, int p)
{
    int res = 1;
    while (k)
    {
        if (k & 1)  res = (LL)res * a % p;
        a = (LL)a * a % p;
        k >>= 1;
    }
    return res;
}

int C(int a, int b, int p)
{
    int res = 1;
    for (int i = 1, j = a; i <= b; i ++, j --)
        res = (LL) res * j % p * quim(i, p - 2, p) % p;
    return res;
}

int lucas(LL a, LL b, int p)
{
    if (a < p && b < p) return C(a, b, p);  //这步也通过别的方法求
    
    return (LL)lucas(a / p, b / p, p) * C(a % p, b % p, p) % p; 
}

int main()
{
    cin >> n;
    while (n --)
    {
        cin >> a >> b >> p;
        cout << lucas(a, b, p) << endl;
    }
}

---

---

四.利用质因数的分解

题目描述

输入a,b，求$C_a^b$的值。

注意结果可能很大，需要使用高精度计算。
1≤b≤a≤5000

问题解法

$C_a^b$ = $a!$ / ($(a-b)!$ * $(b)!$)
可将阶乘分解为质因数的乘积，对于某质因数x，分子的x个数减去分母的x个数便是这个式子的x的个数。最后高精度乘法即可。

参考代码

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;


const int N = 5010;

int primes[N], cnt;
int sum[N];
bool st[N];


void get_primes(int n)
{
    for (int i = 2; i <= n; i ++ )
    {
        if (!st[i]) primes[cnt ++ ] = i;
        for (int j = 0; primes[j] <= n / i; j ++ )
        {
            st[primes[j] * i] = true;
            if (i % primes[j] == 0) break;
        }
    }
}


int get(int n, int p)    //求出n的阶乘中质数p的次数
{                        //即一次求出n/p, n/(p^2), n/(p^3).......
    int res = 0;         //然后将他们相加
    while (n)            //例如，对于某数p^k，他是p, p^2.....p^k的倍数，因此被统计k次
    {                    
        res += n / p;
        n /= p;
    }
    return res;
}


vector<int> mul(vector<int> a, int b)    //高精度乘法
{
    vector<int> c;
    int t = 0;
    for (int i = 0; i < a.size(); i ++ )
    {
        t += a[i] * b;
        c.push_back(t % 10);
        t /= 10;
    }
    while (t)
    {
        c.push_back(t % 10);
        t /= 10;
    }
    return c;
}


int main()
{
    int a, b;
    cin >> a >> b;

    get_primes(a);

    for (int i = 0; i < cnt; i ++ )
    {
        int p = primes[i];
        sum[i] = get(a, p) - get(a - b, p) - get(b, p); //得到C(a, b)的唯一分解
    }

    vector<int> res;
    res.push_back(1);

    for (int i = 0; i < cnt; i ++ )
        for (int j = 0; j < sum[i]; j ++ )
            res = mul(res, primes[i]);    //高精度乘法

    for (int i = res.size() - 1; i >= 0; i -- ) printf("%d", res[i]);
    puts("");

    return 0;
}

---

---

– 思路源自yxc大佬