求组合数

最新推荐文章于 2024-07-14 15:36:20 发布

(￣(工)￣)霸天下

最新推荐文章于 2024-07-14 15:36:20 发布

阅读量49

点赞数

分类专栏：数学知识文章标签： c++ 数据结构开发语言

本文链接：https://blog.csdn.net/m0_64378422/article/details/126155176

版权

数学知识专栏收录该内容

11 篇文章 0 订阅

订阅专栏

一、求组合数 I（递推预处理）

输入样例：

输出样例：

3
10
1

#include<bits/stdc++.h>
using namespace std;
const int N=2010,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]=(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;
}

二、求组合数 II（阶乘逆元预处理）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100010,mod=1e9+7;
int fact[N],infact[N];
int qmi(int a,int b,int p)
{
    int res=1;
    while(b)
    {
    if(b&1)res=(ll)res*a%p;
    a=(ll)a*a%p;
    b>>=1;
    }
    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;
}

三、求组合数Ⅲ（卢卡斯定理）

定理具体推导：

AcWing 887. 求组合数 III(lucas定理) - AcWinghttps://www.acwing.com/solution/content/26553/

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int p;
int qmi(int a,int k)
{
    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 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;
    }
    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;
}

四、求组合数Ⅳ（高精度）

具体:

筛素数(1~5000)
求每个质数的次数
用高精度乘把所有质因子乘上

#include<bits/stdc++.h>
using namespace std;
const int N=5010;
int prime[N],cnt;
int sum[N];
bool st[N];
void get_prime(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(!st[i]) prime[cnt++]=i;
        for(int j=0;prime[j]<=n/i;j++)
        {
            st[prime[j]*i]=true;
            if(i%prime[j]==0)break;
        }
    }
}
int get(int n,int p)
{
    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;
    get_prime(a);
    for(int i=0;i<cnt;i++)
    {
        int p=prime[i];
        sum[i]=get(a,p)-get(b,p)-get(a-b,p);
    }
    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,prime[i]);
    for(int i=res.size()-1;i>=0;i--)
    cout<<res[i];
    return 0;
}