大组合数问题
组合数的公式: C n r = n ∗ ( n − 1 ) ∗ . . . ∗ ( n − r + 1 ) r ! = n ! r ! ( n − r ) ! C^r_n=\frac{n*(n-1)* ... *(n-r+1)}{r!}=\frac{n!}{r!(n-r)!} Cnr=r!n∗(n−1)∗...∗(n−r+1)=r!(n−r)!n!观察上述式子,读者应该发现了,我们主要还是采用第一个算式,因为乘法操作相对较少,速度比较快,也相对来说不容易溢出。
思路一
暴力做乘法,并且使用组合数本身的性质进行优化,使用long long,保证在乘法过程中可以连乘更多的数。另外我们也可以优先做除法,再做乘法。当然这种优化方式能带来的支持运算的范围增大是相当小的。
A.
C
n
0
=
C
n
n
=
1
C^0_n=C^n_n=1
Cn0=Cnn=1
B.
C
n
r
=
C
n
n
−
r
C^r_n=C^{n-r}_n
Cnr=Cnn−r
#include<bits/stdc++.h>
using namespace std;
int main()
{
int n,r;
long long mon=1,son=1;
cin>>n>>r;
if(r==0||r==n) cout<<1;
if(r>n-r) r=n-r;
for(int i=1;i<=r;i++,--n)
{
son*=i;
mon*=n;
}
cout<<mon/son;
return 0;
}
思路二
其实我们就是要想办法让数在连乘过程中不要因为达到上界发生溢出而导致错误,所以我们可以考虑在做乘法的过程中同时除。这样会导致当前没法整除,而在计算机中,整数中的整除是下取整的,所以用整数直接做除法是不正确的。解决办法有三个:
- 我们将除数做缓存,每次可以整除的时候,拿出来除,缩小被除数,并从缓存表中删去这个除数
- 直接用浮点数存,虽然浮点数会有精度损失,但是损失的量是十分小的,而组合数的结果一定是整数,因此我们将浮点结果上取整就好啦~~ ( 但是double类型的存储范围要比long long广,不过需要同时考虑到浮点数本身带来的精度损失,因此我们只当作一种思维拓展,而不把他们视为正规解法 )。
- .我们如果对上面的公式两边取对数,就可以得到 ln ( C n r ) = ln ( n ! ) − ln ( r ! ) − ln ( ( n − r ) ! ) = ∑ i = 1 n ln ( i ) − ∑ i = 1 r ln ( i ) − ∑ i = 1 n − r ln ( i ) \ln(C^r_n)=\ln(n!)-\ln(r!)-\ln((n-r)!)=\sum_{i=1}^n\ln(i)-\sum_{i=1}^r\ln(i)-\sum_{i=1}^{n-r}\ln(i) ln(Cnr)=ln(n!)−ln(r!)−ln((n−r)!)=i=1∑nln(i)−i=1∑rln(i)−i=1∑n−rln(i)使用对数的性质可以帮助我们压缩存储范围,同时改乘除法为加减法,但问题是计算机本身的存储浮点数带来的精度损失问题。
思路三
套大整数的模板。但是大整数采用数组实现,所以速度是比较慢的,另外,有时会碰到取模的要求,在大整数类型中做除法和取模操作是十分不方便的。
思路四
采用杨辉三角递推式。对于组合数,有这么一个性质: C n r = C n − 1 r + C n − 1 r − 1 C^r_n=C^r_{n-1}+C^{r-1}_{n-1} Cnr=Cn−1r+Cn−1r−1我们以取物问题为例,理解上述公式: C n r C_n^r Cnr 表示从n个物体中拿出r个,那么如果我们针对第n个物体考虑,可以将这个原问题划分为两个小的子问题:
- 我们将第n个物体当作我们拿出的第r个物体,此时我们需要在前n-1个物体中拿出r-1个,即 C n − 1 r C^r_{n-1} Cn−1r;
- 我们保证不拿第n个物体,此时我们需要在前n-1个物体中拿出r个,即 C n − 1 r − 1 C^{r-1}_{n-1} Cn−1r−1
#include<bits/stdc++.h>
using namespace std;
int C[1010][1010];
int main()
{
int n,r,i,j;
cin>>n>>r;
C[1][0]=C[1][1]=1;
for(i=2;i<=n;++i)
{
C[i][0]=C[i][i]=1;
for(j=1;j<n;j++) C[i][j]=C[i-1][j]+C[i-1][j-1];
}
cout<<C[n][r];
return 0;
}
相信读者也发现了,递推式本身过于消耗空间,因此递推法能解决的范围也很有限。当然细心的读者应该发现了,这边的递推数组是一个下三角矩阵,因此可以采用坐标转换的方式化为一维存储,可以省去一半的空间。
注意,这个方法需要在打表计算的同时就进行取模操作
(
a
+
b
)
m
o
d
p
=
(
a
m
o
d
p
+
b
m
o
d
p
)
(a+b)\bmod p=(a\mod p+b\mod p)%p
(a+b)modp=(amodp+bmodp),因为在不取模进行累加时,数据的增长是异常迅速的,因此很快就会数据溢出报错。
思路五
用数论的知识,所有的数都可以使用素因数进行分解:即对任意一个数n,存在一个素数序列 P i P_i Pi,使得 n = ∏ i = 1 k P i Q i n=\prod^k_{i=1}Pi^{Qi} n=i=1∏kPiQi我们使用质因数分解的办法,将除数和被除数分解成小的素数因子后,再统计计算结果。因而我们需要一个素数表,我们可以使用欧拉筛快速获得这张表。而也正是因为这样,我们的算法也有了局限性,由于内存的限制,我们没办法获得很大的素数,也就导致了当需要被分解的数中存在大素数因子的时候,我们就束手无测。(或者当素数表使用完之后,继续向后枚举素数,这样会降低算法性能),不过一般情况下,至少处理百万级的因数时,是完全可以的。
#include<bits/stdc++.h>
using namespace std;
const int MaxN = 1000010;
bool isPrime[MaxN];
int Primes[MaxN],cnt,Factor[MaxN];
void EulrSift()//欧拉筛
{
int i,j;
for(i=2;i<=MaxN;i++)
if(!isPrime[i]){
Primes[cnt++]=i;
for(j=0;j<cnt&&Primes[j]*i<MaxN;++j)
{
isPrime[Primes[j]*i]=true;
if(i%Primes[j]) break;
}
}
}
void PrimeResolve(int num,int delta)//质因数分解
{
for(int i=0;num!=1&&i<cnt;i++)
while(num%Primes[i]==0&&num!=1)
{
num/=Primes[i];
Factor[i]+=delta;
}
}
int quickMul(int a,int b)//快速幂
{
int r = 1;
while(b)
{
if(b&1) r*=a;
b>>=1;
a*=a;
}
return r;
}
int main()
{
int n,r,m,i,ans=1;
cin>>n>>r;
m=n;
EulrSift();
for(i=1;i<=r;i++,n--)
{
PrimeResolve(n,1);
PrimeResolve(i,-1);
}
for(i=0;i<cnt&&Primes[i]<=m;i++)
if(Factor[i]!=0 )
ans*=quickMul(Primes[i],Factor[i]);
cout<<ans;
return 0;
}
思路六
从这里开始,我们会接触到求大组合数+取模的问题。观察组合数的运算公式,我们发现运算过程中有除法,但是在取模条件下,我们可以把除法使用逆元的方法转化为乘法来做。费马定理的要求过于苛刻,我们可以采用扩展欧几里得来求。欧拉定理比费马小定理的通用性更强,但是有时候模数过大,可能导致欧拉定理无法使用,另外,对于欧拉定理而言,一般我们不使用他的关键原因在于,本身欧拉定理告诉我们,如果
gcd
(
A
,
P
)
=
1
\gcd(A,P)=1
gcd(A,P)=1,那么一定会有:
A
φ
(
P
)
=
1
(
m
o
d
P
)
A^{\varphi(P)}=1\pmod P
Aφ(P)=1(modP)但是即便我们使用欧拉筛取求解这个
φ
(
P
)
\varphi(P)
φ(P),也需要
O
(
P
)
O(\sqrt P)
O(P)的复杂度,并不值得。但是我们还是复习一遍欧拉函数吧(233)。
我们针对问题:
C
n
r
m
o
d
p
C^r_n\bmod p
Cnrmodp,给出下面两种解决办法:
扩展欧几里得版:
#include<bits/stdc++.h>
using namespace std;
int extgcd(int A,int B,int &x,int &y)
{
if(B==0)
{
x=1;
y=0;
return A;
}
int gcd=extgcd(B,A%B,x,y);
int x0=y;
int y0=x-A/B*y;
x=x0;
y=y0;
return gcd;
}
//A*x=1(mod P)=>A*x+P*y=1
int getInv(int A,int p)
{
int x,y,gcd=extgcd(A,p,x,y);
if(gcd!=1) return -1;//扩展欧几里得无法求出逆元
p/=gcd;
return (x%p+p)%p;//取最小正整数解
}
int main()
{
int i,n,r,p,ans=1,inv;
cin>>n>>r>>p;
for(i=1;i<=r;i++,n--)
{
ans=ans*n%p;
if(ans%r==0) ans/=r;//如果可以除掉,就不要求逆元
else if(r>=2)//1无需求逆元
{
inv=getInv(r,p);
if(inv==-1)
{
cout<<"No solution"<<endl;
return 0;
}
ans=ans*inv%p;
}
}
cout<<ans;
return 0;
}
欧拉函数版:
#include<bits/stdc++.h>
using namespace std;
int cnt,phi[100010],Primes[100010];
int Phi(int n)//质因数分解 结合 欧拉函数性质
{
int res=n;
for(int i=2;i*i<=n;i++)
{
if(n%i==0) res=res/i*(i-1);
while( n % i == 0 ) n/=i;
}
if(n>1)//除干净了因子后 n = 1 或者是最后一个质因数的幂次
{
res=res/n*(n-1);
}
return res;
}
int quickMul(int A,int B,int P)
{
int r=1;
while(B)
{
if(B&1) r=r*A%P;
B>>=1;
A=A*A%P;
}
return r;
}
int gcd(int A,int B)
{
while(B^=A^=B^=A%=B);
return A;
}
int main()
{
int n,r,p,i,ans=1,phi,inv;//son用来存储那些没有逆元的因子
cin>>n>>r>>p;
phi=Phi(p);
for(i=1;i<=r;i++,n--)
{
ans=ans*n ;
if(gcd(i,p)!= 1)
{
cout<<"No solution"<<endl;
return 0;
}
else
{
inv=quickMul(i,phi,p);
ans=ans*inv%p;
}
ans%=p;
}
cout<<ans;
return 0;
}
有些读者可能会想到,本身求逆元是需要时间的,能否在能整除时做除,不能做整除的时候求逆元改为乘法,同时每一步保持取模,答案是否定的,我们不能在乘法、除法、逆元分别取模后再合并。
以
12
3
(
m
o
d
9
)
=
4
\frac{12}{3}\pmod 9=4
312(mod9)=4为例,如果我们做完分子的取模后,再除掉分母后再取模,那么答案为:
12
(
m
o
d
9
)
3
(
m
o
d
9
)
=
3
3
(
m
o
d
9
)
=
1
\frac{12 \pmod9}{3}\pmod9=\frac{3}{3}\pmod9=1
312(mod9)(mod9)=33(mod9)=1原因在于,取模本身不满足和除法得交换律:
- 因此,我们只能选择,在可以整除的时候,整除,但是一旦除法参与了,那么取模运算必须放到最后进行。
- 全部改用逆元做乘法。
我们使用这些算法的目的就是为了避免乘法溢出,因此方案1肯定是不切实际的。
思路7:卢卡斯定理
卢卡斯定理提供了一种在模数为素数的情况下,由于n和r远大于p,所以在除法过程中求逆元的时候很容易遇到与p不满足互质的情况,例如2p,3p,从而由于不满足互质导致无法求解逆元,但是卢卡斯定理提供了一种这种情形下还能求解大组合数的方案。
定理内容:对于非负整数n,r,p(其中p为素数),那么有如下的同余方程:
C
n
r
=
∏
i
=
0
k
C
n
i
r
i
(
m
o
d
p
)
C^r_n=\prod^k_{i=0}C^{r_i}_{n_i} \pmod p
Cnr=i=0∏kCniri(modp)其中,
n
i
n_i
ni,
r
i
r_i
ri需要满足如下等式:
n
k
p
k
+
n
k
−
1
p
k
−
1
+
.
.
.
+
n
1
p
+
n
0
n_kp^k+n_{k-1}p^{k-1}+...+n_1p+n_0
nkpk+nk−1pk−1+...+n1p+n0
r
k
p
k
+
r
k
−
1
p
k
−
1
+
.
.
.
+
r
1
p
+
r
0
r_kp^k+r_{k-1}p^{k-1}+...+r_1p+r_0
rkpk+rk−1pk−1+...+r1p+r0不难发现,这就是分别将n,r做p进制下的分解罢了。
在实际的算法实现上,我们会这么使用这个定理,与上述是等价的:
L
u
c
a
s
n
r
=
C
n
m
o
d
p
r
m
o
d
p
L
u
c
a
s
n
/
p
r
/
p
Lucas^r_n=C^{r\bmod p}_{n\bmod p}Lucas^{r/p}_{n/p}
Lucasnr=CnmodprmodpLucasn/pr/p注意上述公式中的除法,是整除,其实上述公式完全等价于卢卡斯定理原本的公式。
需要注意的是,在这个定理中,如果发生了
n
<
r
n<r
n<r的情况,那么规定
C
n
r
=
0
C_n^r=0
Cnr=0。
公式证明:求解
C
p
x
(
m
o
d
p
)
C^x_p\pmod p
Cpx(modp),设
x
<
p
x<p
x<p,
x
≠
0
x≠0
x=0
C
p
x
=
p
!
x
!
(
p
−
x
)
!
=
p
∗
(
p
−
1
)
!
x
∗
(
x
−
1
)
!
∗
(
p
−
x
)
!
=
p
x
∗
C
p
−
1
x
−
1
C^x_p=\frac{p!}{x!(p-x)!}=\frac{p*(p-1)!}{x*(x-1)!*(p-x)!}=\frac{p}{x}*C^{x-1}_{p-1}
Cpx=x!(p−x)!p!=x∗(x−1)!∗(p−x)!p∗(p−1)!=xp∗Cp−1x−1由于我们已经设
x
<
p
x<p
x<p且p为素数,因此我们知道x肯定有关于p的逆元,所以上述式子可以转换为:
C
p
x
(
m
o
d
p
)
=
p
x
∗
C
p
−
1
x
−
1
(
m
o
d
p
)
=
p
∗
i
n
v
(
x
∣
p
)
∗
C
n
−
1
r
−
1
(
m
o
d
p
)
C^x_p\pmod p=\frac{p}{x}*C^{x-1}_{p-1}\pmod p=p*inv(x\mid p)*C^{r-1}_{n-1} \pmod p
Cpx(modp)=xp∗Cp−1x−1(modp)=p∗inv(x∣p)∗Cn−1r−1(modp)
i
n
v
(
x
∣
p
)
inv(x\mid p)
inv(x∣p)表示x在模p下的逆元。
此时等式右端是p的倍数,因此
C
p
x
=
0
(
m
o
d
p
)
C_p^x=0\pmod p
Cpx=0(modp)。
根据二项式定理:
(
1
+
x
)
n
=
∑
i
=
0
n
C
n
i
∗
x
i
(1+x)^n=\sum^n_{i=0}C^i_n*x^i
(1+x)n=i=0∑nCni∗xi我们知道:
(
1
+
x
)
p
=
∑
i
=
0
p
C
p
i
∗
x
i
(1+x)^p=\sum^p_{i=0}C^i_p*x^i
(1+x)p=i=0∑pCpi∗xi根据我们上面所证,这个上述式子的展开式中,只有
i
=
0
i=0
i=0和
i
=
p
i=p
i=p的两项模p不为0。故我们得到:
(
1
+
x
)
p
=
1
+
x
p
(
m
o
d
p
)
(1+x)^p=1+x^p\pmod p
(1+x)p=1+xp(modp)我们先预设:
{
n
/
p
=
n
k
n
m
o
d
p
=
r
e
s
k
\begin{cases}n/p=n_k\\n \bmod p=res_k\end{cases}
{n/p=nknmodp=resk我们对
(
1
+
x
)
n
(1+x)^n
(1+x)n做p进制分解:
(
1
+
x
)
n
=
(
1
+
x
)
n
k
∗
p
+
r
e
s
k
=
(
1
+
x
)
n
k
∗
p
+
(
1
+
x
)
r
e
s
k
=
[
(
1
+
x
)
p
]
n
k
∗
(
1
+
x
)
r
e
s
k
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
(
1
)
=
(
1
+
x
p
)
n
k
∗
(
1
+
x
)
r
e
s
k
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
(
2
)
=
∑
i
=
0
n
k
C
n
k
i
∗
(
x
p
)
i
∗
∑
j
=
0
r
e
s
k
∗
x
j
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
.
(
3
)
=
∑
i
=
0
n
k
∑
j
=
0
r
e
s
k
C
n
k
i
∗
C
r
e
s
k
j
∗
x
i
p
+
j
(
m
o
d
p
)
.
.
.
.
(
4
)
\begin{aligned}(1+x)^n&=(1+x)^{n_k*p+res_k}\\&=(1+x)^{n_k*p}+(1+x)^{res_k}\\&=[(1+x)^p]^{n_k}*(1+x)^{res_k}......................(1)\\&=(1+x^p)^{n_k}*(1+x)^{res_k}........................(2)\\&=\sum^{n_k}_{i=0}C^i_{n_k}*(x^p)^i*\sum^{res_k}_{j=0}*x^j.......................(3)\\&=\sum^{n_k}_{i=0}\sum^{res_k}_{j=0}C^i_{n_k}*C^j_{res_k}*x^{ip+j}\pmod p....(4)\end{aligned}
(1+x)n=(1+x)nk∗p+resk=(1+x)nk∗p+(1+x)resk=[(1+x)p]nk∗(1+x)resk......................(1)=(1+xp)nk∗(1+x)resk........................(2)=i=0∑nkCnki∗(xp)i∗j=0∑resk∗xj.......................(3)=i=0∑nkj=0∑reskCnki∗Creskj∗xip+j(modp)....(4)(1)->(2):在%P的条件下,我们用了上述推出的结论
(
1
+
x
)
p
=
1
+
x
p
(
m
o
d
p
)
(1+x)^p=1+x^p \pmod p
(1+x)p=1+xp(modp)
(2)->(3):使用两次二项式展开定理
(3)->(4):多项式的乘法就是分别乘法后求和
注意(2)式中我们用了上面推导的结论:
(
1
+
x
)
p
=
1
+
x
p
(
m
o
d
p
)
(1+x)^p=1+x^p\pmod p
(1+x)p=1+xp(modp)
所以我们得到了:
(
1
+
x
)
p
=
∑
r
=
0
n
C
n
r
∗
x
r
=
∑
i
=
0
n
k
C
n
k
i
x
i
p
∗
∑
j
=
0
r
e
s
k
C
r
e
s
k
j
x
j
=
∑
i
=
0
n
k
∑
j
=
0
r
e
s
k
C
n
k
i
C
r
e
s
k
j
x
i
p
+
j
(
m
o
d
p
)
,
(
x
<
p
,
r
e
s
k
<
p
)
\begin{aligned}(1+x)^p&=\sum^n_{r=0}C^r_n*x^r\\&=\sum^{n_k}_{i=0}C^i_{n_k}x^{ip}*\sum^{res_k}_{j=0}C^j_{res_k}x^j\\&=\sum^{n_k}_{i=0}\sum^{res_k}_{j=0}C^i_{n_k}C^j_{res_k}x^{ip+j}\pmod p,(x<p,res_k<p)\end{aligned}
(1+x)p=r=0∑nCnr∗xr=i=0∑nkCnkixip∗j=0∑reskCreskjxj=i=0∑nkj=0∑reskCnkiCreskjxip+j(modp),(x<p,resk<p)由于上述式是恒等式,因此比较两端
x
k
x^k
xk的系数可知:
C
n
r
x
r
=
C
n
k
i
C
r
e
s
k
j
x
i
p
+
j
(
m
o
d
p
)
,
(
x
<
p
,
r
e
s
k
<
p
)
C^r_nx^r=C^i_{n_k}C^j_{res_k}x^{ip+j}\pmod p,(x<p,res_k<p)
Cnrxr=CnkiCreskjxip+j(modp),(x<p,resk<p)即:
C
n
r
=
C
n
k
i
C
r
e
s
k
j
(
m
o
d
p
)
,
(
x
<
p
,
r
e
s
k
<
p
)
C^r_n=C^i_{n_k}C^j_{res_k}\pmod p,(x<p,res_k<p)
Cnr=CnkiCreskj(modp),(x<p,resk<p)
n
=
p
n
k
+
r
e
s
k
n=pn_k+res_k
n=pnk+resk
r
=
i
p
+
j
r=ip+j
r=ip+j则
C
n
r
=
C
n
k
i
C
r
e
s
k
j
=
C
n
/
p
r
/
p
C
n
m
o
d
p
r
m
o
d
p
C^r_n=C^i_{n_k}C^j_{res_k}=C^{r/p}_{n/p}C^{r\bmod p}_{n\bmod p}
Cnr=CnkiCreskj=Cn/pr/pCnmodprmodp证毕。由于
r
m
o
d
p
<
p
r\bmod p<p
rmodp<p,
n
m
o
d
p
<
p
n\bmod p<p
nmodp<p,因此对于
C
n
m
o
d
p
r
m
o
d
p
C_{n\bmod p}^{r\bmod p}
Cnmodprmodp我们就可以比较容易地计算(因为p是质数,所以对与
x
<
p
x<p
x<p,一定有
gcd
(
x
,
p
)
=
1
\gcd(x,p)=1
gcd(x,p)=1,互质条件下可以求逆元)
在卢卡斯定理的加持下,我们可以保证在如下式子中:
C
n
r
=
∏
i
=
0
k
C
n
i
r
i
(
m
o
d
p
)
C^r_n=\prod^k_{i=0}C^{r_i}_{n_i}\pmod p
Cnr=i=0∏kCniri(modp)一定会有
n
i
,
r
i
<
p
n_i,r_i<p
ni,ri<p(按照p进制数理解,每一位上的权值不可能大于它所表示的数的进制值)。
而又由于p为素数,因此我们又可以用逆元啦,完美!
#include<bits/stdc++.h>
using namespace std;
int quickMul(int A,int B,int P)
{
int res=1;
while(B)
{
if(B&1) res=res*A%P;
B>>=1;
A=A*A%P;
}
return res;
}
int inv(int n,int p)// 费马小定理求解逆元 n^(p-1)=1(mod p)
{
return quickMul(n,p-2,p);
}
int C(int n,int r,int p)
{
int res=1;
for(int i=1;i<=r;i++,n--) res=res*n%p*inv(i,p)%p;
return res;
}
int Lucas(int n,int r,int p)
{
return r==0?1:Lucas(n/p,r/p,p)*C(n%p,r%p,p)%p;
}
int main()
{
int n,r,p;
cin>>n>>r>>p;
if(n<r||n==0)
{
cout<<-1;// 视具体题目要求而定,毕竟数学上规定 0!=1
return 0;
}
if(p==1)
{
cout<<0;
return 0;
}
cout<<Lucas(n,r,p);
return 0;
}
来看一道模板题吧~
洛谷 P3807 【模板】卢卡斯定理
题目背景
这是一道模板题。
题目描述
给定整数
n
,
m
,
p
n,m,p
n,m,p 的值,求出
C
n
+
m
n
m
o
d
p
C^n_{n+m}\bmod p
Cn+mnmodp 的值。
输入数据保证
p
p
p为质数。
注:
C
C
C 表示组合数。
输入格式
本题有多组数据。
第一行一个整数
T
T
T,表示数据组数。
对于每组数据:
一行,三个整数
n
n
n,
m
m
m,
p
p
p。
输出格式
对于每组数据,输出一行,一个整数,表示所求的值。
输入输出样例
输入#1
2
1 2 5
2 1 5
输出#1
3
3
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll quickMul(ll A,ll B,ll P)
{
ll res = 1;
while(B)
{
if(B&1) res=res*A%P;
B>>=1;
A=A*A%P;
}
return res;
}
ll inv(ll n,ll p)// 费马小定理求解逆元 n^(p-1)=1(mod p)
{
return quickMul(n,p-2,p);
}
ll C(ll n,ll r,ll p)
{
ll res=1;
for(int i=1;i<=r;++i,--n) res=res*n%p*inv(i,p)%p;
return res;
}
ll Lucas(ll n,ll r,ll p)
{
return r==0?1:Lucas(n/p,r/p,p)*C(n%p,r%p,p)%p;
}
int main()
{
ll T,n,m,p;
cin>>T;
while(T--)
{
cin>>n>>m>>p;
n+=m;
if(p==1) cout<<0<<endl;
else cout<<Lucas(n,m,p)<<endl;
}
return 0;
}
奇技淫巧:
在使用卢卡斯定理的时候,我们还可以预处理阶乘,这样可以加快速度,在预处理阶乘时,我们采用公式二:
C
n
r
=
n
!
r
!
(
n
−
r
)
!
C^r_n=\frac{n!}{r!(n-r)!}
Cnr=r!(n−r)!n!事实上我们可以得到:
C
n
r
(
m
o
d
p
)
=
n
!
r
!
(
n
−
r
)
!
(
m
o
d
p
)
=
n
!
(
m
o
d
p
)
∗
i
n
v
(
r
!
∣
p
)
(
m
o
d
p
)
∗
i
n
v
(
n
−
r
!
(
m
o
d
p
)
∣
p
)
(
m
o
d
p
)
=
n
!
(
m
o
d
p
)
∗
i
n
v
(
r
!
(
m
o
d
p
)
∣
p
)
(
m
o
d
p
)
∗
i
n
v
(
n
−
r
!
(
m
o
d
p
)
∣
p
)
(
m
o
d
p
)
\begin{aligned}C^r_n\pmod p&=\frac{n!}{r!(n-r)!}\pmod p\\&=n!\pmod p*inv(r!\mid p)\pmod p*inv(n-r!\pmod p\mid p)\pmod p\\&=n!\pmod p*inv(r!\pmod p\mid p)\pmod p*inv(n-r!\pmod p\mid p)\pmod p\end{aligned}
Cnr(modp)=r!(n−r)!n!(modp)=n!(modp)∗inv(r!∣p)(modp)∗inv(n−r!(modp)∣p)(modp)=n!(modp)∗inv(r!(modp)∣p)(modp)∗inv(n−r!(modp)∣p)(modp)即我们需要证明
x
!
y
!
(
m
o
d
p
)
=
x
!
∗
i
n
v
(
y
!
(
m
o
d
p
)
∣
p
)
(
m
o
d
p
)
\frac{x!}{y!}\pmod p=x!*inv(y!\pmod p\mid p)\pmod p
y!x!(modp)=x!∗inv(y!(modp)∣p)(modp)
设
z
=
y
!
(
m
o
d
p
)
z=y!\pmod p
z=y!(modp),即
y
=
z
+
k
p
y=z+kp
y=z+kp,那么有
z
<
p
z<p
z<p,且p为素数,故z必有逆元,则有:
z
∗
i
n
v
(
z
∣
p
)
=
1
(
m
o
d
p
)
z*inv(z\mid p)=1\pmod p
z∗inv(z∣p)=1(modp)那么
y
!
∗
i
n
v
(
z
)
=
(
k
p
+
z
)
i
n
v
(
z
∣
p
)
=
z
∗
i
n
v
(
z
)
+
k
p
∗
i
n
v
(
z
∣
p
)
=
z
∗
i
n
v
(
z
)
(
m
o
d
p
)
=
1
\begin{aligned}y!*inv(z)&=(kp+z)inv(z\mid p)\\&=z*inv(z)+kp*inv(z\mid p)\\&=z*inv(z)\pmod p\\&=1\end{aligned}
y!∗inv(z)=(kp+z)inv(z∣p)=z∗inv(z)+kp∗inv(z∣p)=z∗inv(z)(modp)=1证毕。
#include<bits/stdc++.h>
using namespace std;
const int MaxN=1e5+10;
typedef long long ll;
ll Factorial[MaxN<<1];
ll quickMul(ll A,ll B,ll P)
{
ll res=1;
while(B)
{
if(B&1) res=res*A%P;
B>>=1;
A=A*A%P;
}
return res;
}
ll inv(ll n,ll p)
{
return quickMul(n,p-2,p);
}
ll C(ll n,ll r,ll p)// 这里需要注意 n < r 时 是非法状态 返回 0
{
return n<r?0:Factorial[n]*inv(Factorial[r],p)%p*inv(Factorial[n-r],p)%p;
}
ll Lucas(ll n,ll r,ll p)
{
return r==0?1:Lucas(n/p,r/p,p)*C(n%p,r%p,p)%p;
}
int main()
{
ll T,n,m,p,i;
cin>>T;
Factorial[1]=1;
while(T--)
{
cin>>n>>m>>p;
n+=m;
memset(Factorial,0,sizeof(Factorial));
Factorial[1]=1;
if(p==1) cout<<0<<endl;
else
{
m=m>(n>>1)?n-m:m;
for(i=2;i<=n;i++) Factorial[i]=Factorial[i-1]*i%p;
cout<<Lucas(n,m,p)<<endl;
}
}
return 0;
}
————————————————
版权声明:本文为CSDN博主「TMZT」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_39304630/article/details/77417017
有更改
思路八:扩展Lucas定理
在Lucas定理中,模数只能是质数。但很多情况下,模数不一定是质数。这个时候就需要用到Lucas定理的扩展版——扩展Lucas定理。
扩展卢卡斯定理用于求如下式子(其中p不一定是质数):
C
m
n
m
o
d
p
C^n_m \bmod p
Cmnmodp
我们将这个问题由总体到局部地分为三个层次解决。
层次一:原问题
首先对p进行质因数分解: p = ∏ i p i k i p=\prod_ip_i^{k_i} p=i∏piki显然 p i k i p_i^{k_i} piki 是两两互质的,所以如果分别求出 C m n m o d p i k i C^n_m \bmod p_i^{k_i} Cmnmodpiki,就可以构造出若干个形如 C n m = a i m o d p i k i C^m_n=a_i\bmod p_i^{k_i} Cnm=aimodpiki的方程,然后用中国剩余定理即可求解。
层次二:组合数模质数幂
现在的问题就转化成了求如下式子(其中p是质数): C n m m o d p k C^m_n\bmod p^k Cnmmodpk脑补一下组合数公式 C n m = n ! m ! ( n − m ) ! C^m_n=\frac{n!}{m!(n-m)!} Cnm=m!(n−m)!n!,发现由于 m ! m! m! 和 ( n − m ) ! (n−m)! (n−m)! 可能包含质因子p,所以不能直接求他们对于 p k p^k pk 的逆元。此时我们可以将 n ! n! n! 、 m ! m! m!、 ( n − m ) ! (n-m)! (n−m)! 中的质因子 p p p 全部提出来,最后再乘回去即可。即变为下式( k 1 k1 k1 为 n ! n! n! 中质因子 p p p 的次数, k 2 k2 k2、 k 3 k3 k3同理): n ! p k 1 m ! p k 2 × ( n − m ) ! p k 3 × p p 1 − p 2 − p 3 \cfrac{\cfrac{n!}{p^{k1}}}{\cfrac{m!}{p^{k2}}×\cfrac{(n-m)!}{p^{k3}}}×p^{p1-p2-p3} pk2m!×pk3(n−m)!pk1n!×pp1−p2−p3 m ! p k 2 \frac{m!}{p^{k2}} pk2m! 和 ( n − m ) ! p k 3 \frac{(n-m)!}{p^{k3}} pk3(n−m)! 和 p k p^k pk 是互质的,可以直接求逆元。
层次三:阶乘除去质因子后模质数幂
现在看看如何计算形如下式的式子。
n
!
p
a
m
o
d
p
k
\frac{n!}{p^a}\bmod p^k
pan!modpk先考虑如何计算
n
!
m
o
d
p
k
n!\bmod p^k
n!modpk。
举个例子:n=22,p=3,k=2
把这个写出来:
22
!
=
1
×
2
×
3
×
4
×
5
×
6
×
7
×
8
×
9
×
10
×
11
×
12
×
13
×
14
×
15
×
16
×
17
×
18
×
19
×
20
×
21
×
22
22 ! = 1 × 2 × 3 × 4 × 5 × 6 × 7 × 8 × 9 × 10 × 11 × 12 × 13 × 14 × 15 × 16 × 17 × 18 × 19 × 20 × 21 × 22
22!=1×2×3×4×5×6×7×8×9×10×11×12×13×14×15×16×17×18×19×20×21×22
把其中所有p(也就是3)的倍数提取出来,得到:
22
!
=
3
7
×
(
1
×
2
×
3
×
4
×
5
×
6
×
7
)
×
(
1
×
2
×
4
×
5
×
7
×
8
×
10
×
11
×
13
×
14
×
16
×
17
×
19
×
20
×
22
)
22!=3^7×(1×2×3×4×5×6×7)×(1×2×4×5×7×8×10×11×13×14×16×17×19×20×22)
22!=37×(1×2×3×4×5×6×7)×(1×2×4×5×7×8×10×11×13×14×16×17×19×20×22)
可以看出上式分为三个部分:第一个部分是3的幂,次数是小于等于22的3的倍数的个数,即
⌊
n
p
⌋
\lfloor\frac{n}{p}\rfloor
⌊pn⌋ 。
第二个部分是一个阶乘7!,即
⌊
n
p
⌋
!
\lfloor\frac{n}{p}\rfloor!
⌊pn⌋!,可以递归解决。
第三个部分是n!中与p互质的部分的乘积,这一部分具有如下性质:
1
×
2
×
4
×
5
×
7
×
8
≡
10
×
11
×
13
×
14
×
16
×
17
(
m
o
d
p
k
)
1\times2\times4\times5\times7\times8\equiv10\times11\times13\times14\times16\times17 \pmod {p^k}
1×2×4×5×7×8≡10×11×13×14×16×17(modpk)
在模
3
2
3^2
32 的意义下10和1同余,11和2同余……写成下式就比较显然。(t是任意正整数)
∏
i
,
(
i
,
p
)
=
1
p
k
i
≡
∏
i
,
(
i
,
p
)
=
1
p
k
(
i
+
t
p
k
)
(
m
o
d
p
k
)
\prod^{p^k}_{i,(i,p)=1}i\equiv\prod^{p^k}_{i,(i,p)=1}(i+tp^k)\pmod {p^k}
i,(i,p)=1∏pki≡i,(i,p)=1∏pk(i+tpk)(modpk)
∏
i
,
(
i
,
p
)
=
1
p
k
i
\prod^{p^k}_{i,(i,p)=1}i
∏i,(i,p)=1pki 一共循环了
⌊
n
p
k
⌋
\lfloor\frac{n}{p^k}\rfloor
⌊pkn⌋ 次,暴力求出
∏
i
,
(
i
,
p
)
=
1
p
k
i
\prod^{p^k}_{i,(i,p)=1}i
∏i,(i,p)=1pki ,然后用快速幂求它的
∏
i
,
(
i
,
p
)
=
1
p
k
i
\prod^{p^k}_{i,(i,p)=1}i
∏i,(i,p)=1pki 次幂。
最后还要乘上19×20×22(即
∏
i
,
(
i
,
p
)
=
1
n
m
o
d
p
k
\prod^{n\bmod p^k}_{i,(i,p)=1}
∏i,(i,p)=1nmodpk),显然这一段的长度一定小于
p
k
p^k
pk ,暴力乘上去即可。
如上三部分的乘积就是
n
!
n!
n!。最终要求的是
n
!
p
a
m
o
d
p
k
\frac{n!}{p^a} \bmod p^k
pan!modpk ,分母全部由上述第一部分和第二部分贡献(第三部分和p互质)。而递归计算第二部分的时候已经除去了第二部分中的因子p,所以最终的答案就是上述第二部分递归返回的结果和第三部分的乘积(与第一部分无关)。
结合代码方便理解:
ll fac(const ll n,const ll p,const ll pk)
{
if (!n)
return 1;
ll ans=1;
for(int i=1;i<pk;i++)
if(i%p)
ans=ans*i%pk;
ans=power(ans,n/pk,pk);
for(int i=1;i<=n%pk;i++)
if(i%p)
ans=ans*i%pk;
return ans*fac(n/p,p,pk)%pk;
}
层次二:组合数模质数幂
回到这个式子 n ! p k 1 m ! p k 2 × ( n − m ) ! p k 3 × p p 1 − p 2 − p 3 \cfrac{\cfrac{n!}{p^{k1}}}{\cfrac{m!}{p^{k2}}×\cfrac{(n-m)!}{p^{k3}}}×p^{p1-p2-p3} pk2m!×pk3(n−m)!pk1n!×pp1−p2−p3可以很容易地把它转换成代码(注意i要开long long):
ll C(const ll n,const ll m,const ll p,const ll pk)
{
if(n<m) return 0;
ll f1=fac(n,p,pk),f2=fac(m,p,pk),f3=fac(n-m,p,pk),cnt=0;
for(ll i=n;i;i/=p)
cnt+=i/p;
for(ll i=m;i;i/=p)
cnt-=i/p;
for (ll i=n-m;i;i/=p)
cnt-=i/p;
return f1*inv(f2,pk)%pk*inv(f3,pk)%pk*power(p,cnt,pk)%pk;
}
层次一:原问题
完整代码(题目:洛谷4720【模板】扩展卢卡斯):
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6;
typedef long long ll;
ll n, m, p;
ll power(ll a,ll b,ll p=0x7fffffffffffffff)
{
ll ans=1;
while(b)
{
if(b&1) ans=ans*a%p;
a=a*a%p;
b>>=1;
}
return ans;
}
ll fac(ll n,ll p,ll pk)
{
if(!n) return 1;
ll ans=1;
for(int i=1;i<pk;i++) if(i%p) ans=ans*i%pk;
ans=power(ans,n/pk,pk);
for(int i=1;i<=n%pk;i++) if(i%p) ans=ans*i%pk;
return ans*fac(n/p,p,pk)%pk;
}
ll exgcd(ll a,ll b,ll &x,ll &y)
{
if(!b)
{
x=1;
y=0;
return a;
}
ll xx,yy,g=exgcd(b,a%b,xx,yy);
x=yy;
y=xx-a/b*yy;
return g;
}
ll inv(const ll a, const ll p)
{
ll x,y;
exgcd(a,p,x,y);
return (x%p+p)%p;
}
ll C(ll n,ll m,ll p,ll pk)
{
if(n<m) return 0;
ll f1=fac(n,p,pk),f2=fac(m,p,pk),f3=fac(n-m,p,pk),cnt=0;
for(ll i=n;i;i/=p) cnt+=i/p;
for(ll i=m;i;i/=p) cnt-=i/p;
for(ll i=n-m;i;i/=p) cnt-=i/p;
return f1*inv(f2,pk)%pk*inv(f3,pk)%pk*power(p,cnt,pk)%pk;
}
ll a[N], c[N],cnt;
ll crt()
{
ll M=1,ans=0;
for(int i=0;i<cnt;i++) M*=c[i];
for(int i=0;i<cnt;i++) ans=(ans+a[i]*(M/c[i])%M*inv(M/c[i],c[i])%M)%M;
return ans;
}
ll exlucas(ll n,ll m,ll p)
{
ll tmp=sqrt(p);
for(int i=2;p>1&&i<=tmp;i++)
{
ll tmp=1;
while(p%i==0)
{
p/=i;
tmp*=i;
}
if(tmp>1)
{
a[cnt]=C(n,m,i,tmp);
c[cnt++]=tmp;
}
}
if(p>1)
{
a[cnt]=C(n,m,p,p);
c[cnt++]=p;
}
return crt();
}
int main()
{
cin>>n>>m>>p;
cout<<exlucas(n,m,p);
return 0;
}
————————————————
版权声明:本文为CSDN博主「Inspector_Javert」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/hqddm1253679098/article/details/82897638
有改动