Acwing_887（求组合数Ⅲ）

最新推荐文章于 2024-06-23 13:27:08 发布

笨笨sg

最新推荐文章于 2024-06-23 13:27:08 发布

阅读量364

点赞数 9

分类专栏： Acwing算法基础课文章标签：组合数卢卡斯定理快速幂

本文链接：https://blog.csdn.net/a131529/article/details/136751394

版权

Acwing算法基础课专栏收录该内容

107 篇文章 1 订阅

订阅专栏

本文介绍了卢卡斯定理在计算组合数中的应用，特别是在模素数情况下的高效算法，包括C函数和qmi函数的实现以及递归函数lucas的定义，通过示例展示了如何在AcWing题库中解决求组合数的问题。

摘要由CSDN通过智能技术生成

原题链接：

887. 求组合数 III - AcWing题库

题解：

卢卡斯定理（Lucas' Theorem）是一种用于计算组合数（组合数也称为二项系数）的定理，它在模素数的情况下特别有用。该定理说明了如何将大的组合数分解成小的组合数，从而在求解组合数时可以更高效地利用模运算。

详细推导证明过程参见：AcWing 887. 求组合数 III(lucas定理) - AcWing

代码：

#include<bits/stdc++.h>
using namespace std;
using LL = long long;

int qmi(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
{
	if (b > a)return 0;//漏了边界条件
	int res = 1;
	// a!/(b!(a-b)!) = (a-b+1)*...*a / b! 分子有b项
	for (int i = 1, j = a;i <= b;i++, j--)//i<=b而不是<
	{
		res = (LL)res * j % p;
		res = (LL)res * qmi(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);//lucas递归终点是C_{bk}^{ak}
	return (LL)C(a % p, b % p, p) * lucas(a / p, b / p, p) % p;//a%p后肯定是<p的,所以可以用C(),但a/p后不一定<p 所以用lucas继续递归
}

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