组合数模板-CSDN博客

本文链接：https://blog.csdn.net/lbdd123/article/details/127554241

组合数模板

组合数1

题目大意: 给定 $n$ 组询问，每组询问2给定两个整数 $a, b$ ，请你输出 $C_{a}^b mod(10^9 +7)$ 的值。

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

输出格式
共 $n$ 行，每行输出一个询问的解。
数据范围
$1\leq n\leq 10000$
$1\leq b\leq a\leq 2000$

输入样例:

输出样例

3
10
1

代码:

#include <iostream>

using namespace std;

const int N = 2005;
typedef long long ll;
const int mod = 1e9+7;

int c[N][N];

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

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

组合数2

题目大意: 给定 $n$ 组询问，每组询问2给定两个整数 $a, b$ ，请你输出 $C_{a}^b mod(10^9 +7)$ 的值。

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

输出格式
共 $n$ 行，每行输出一个询问的解。
数据范围
$1\leq n\leq 10000$
$1\leq b\leq a\leq 10^5$
输入样例:

输出样例

3
10
1

代码:

#include <iostream>

using namespace std;

const int N = 100005;
const int mod = 1e9+7;
int fact[N];
int infact[N];

typedef long long ll;

int qmi(int a,int k,int p)
{
    ll res=1;
    while(k)
    {
        if(k&1)res=res*a%p;
        k>>=1;
        a=(ll)a*a%p;
    }
    return res;
}

int main()
{
    fact[0]=infact[0]=1;
    for(int i=1;i<N;i++)
    {
        fact[i]=(ll)fact[i-1]*i%mod;
        infact[i]=(ll)infact[i-1]*qmi(i,mod-2,mod)%mod;
    }
    int n;
    cin>>n;
    while(n--)
    {
        int a,b;
        cin>>a>>b;
        cout<<(ll)fact[a]*infact[b]%mod*infact[a-b]%mod<<endl;
    }
    return 0;
}

组合数3（lucas)

给定 $n$ 组询问，每组询问给定三个整数 $a, b, p$ ，其中 $p$ 是质数，请你输出 $C_{a}^b \ mod\ p$ 的值。

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

数据范围
$1\leq n\leq 20,$
$1\leq b\leq a\leq 10^{18}$
$1\leq p \leq 10^5，$

输入样例:

输出样例:

3
3
2

代码:

#include <iostream>

using namespace std;

typedef long long ll;

int p;

int qmi(int a,int k,int p)
{
    ll res=1;
    while(k)
    {
        if(k&1)res=res*a%p;
        k>>=1;
        a=(ll)a*a%p;
    }
    return res;
}

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

int lucas(ll a,ll b)
{
    if(a<p&&b<p)return C(a,b);
    return (ll)C(a%p,b%p)*lucas(a/p,b/p)%p;
}

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

组合数4

输入 $a, b$ ,求 $C_{a}^b$ 的值。
注意结果可能很大，需要使用高精度计算。

输入格式
共一行，包含两个整数 $a$ 和 $b$ 。
输出格式
共一行，输出 $C_{a}^b$ 的值。
数据范围
$1\leq b\leq a\leq 5000$
输入样例:

5 3

输出样例:

代码:

#include <iostream>
#include <vector>

using namespace std;
const int N = 5005;

int primes[N],sum[N];
bool st[N];
int cnt;
int p;
void prime(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(!st[i])
        {
            primes[cnt++]=i;
        }
        for(int j=0;primes[j]*i<=n;j++)
        {
            st[primes[j]*i]=true;
            if(i%primes[j]==0)break;
        }
    }
}

int get(int n)
{
    int res=0;
    while(n)
    {
        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;
    prime(a);
    for(int i=0;i<cnt;i++)
    {
        p=primes[i];
        sum[i]=get(a)-get(b)-get(a-b);
    }
    vector<int> res;
    res.push_back(1);
    for(int i=0;i<cnt;i++)
    {
        p=primes[i];
        for(int j=0;j<sum[i];j++)
        {
            res=mul(res,p);
        }
    }
    for(int i=res.size()-1;i>=0;i--)
    cout<<res[i];
    return 0;
}