c++求组合数

简简万

于 2024-07-31 11:11:45 发布

阅读量555

点赞数 27

分类专栏：算法基础文章标签： c++ 算法

本文链接：https://blog.csdn.net/wan_guanghui/article/details/140816336

版权

算法基础专栏收录该内容

3 篇文章 0 订阅

订阅专栏

一、组合数的定义

二、预处理逆元求组合数

1.核心

int C(int a,int b)//求组合数 
{
	if(b>a)
	return 0;
	return fact[a]*infact[b]%mod*infact[a-b]%mod;
}

2.阶乘及逆元

求阶乘：

a!=(a−1)!∗a

fact[a] = fact[a - 1] * a

求阶乘的逆元：由费马小定理，在模 p下

int qmi(int a,int b)//快速幂模版 
{
	int ans=1;
	while(b)
	{
		if(b&1)
		ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}
void init()//预处理求阶乘及逆元 
{
	fact[0]=infact[0]=1;
	for(int i=1;i<N;i++)
	{
		fact[i]=fact[i-1]*i%mod;//阶乘 
		infact[i]=infact[i-1]*qmi(i,mod-2)%mod;//阶乘逆元 
	}
}

三、代码模版

#include<iostream>
using namespace std;
#define int long long
const int mod = 1e9+7;
const int N = 1e5+7;
int fact[N],infact[N];//阶乘，阶乘逆元 
int qmi(int a,int b)//快速幂模版 
{
	int ans=1;
	while(b)
	{
		if(b&1)
		ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}
void init()//预处理求阶乘及逆元 
{
	fact[0]=infact[0]=1;
	for(int i=1;i<N;i++)
	{
		fact[i]=fact[i-1]*i%mod;//阶乘 
		infact[i]=infact[i-1]*qmi(i,mod-2)%mod;//阶乘逆元 
	}
}
int C(int a,int b)//求组合数 
{
	if(b>a)
	return 0;
	return fact[a]*infact[b]%mod*infact[a-b]%mod;
}
signed main()
{
	init();
	int a,b;
	cin>>a>>b;
	cout<<C(a,b);
	
}