hdu 3037 費馬小定理+逆元求組合數+Lucas定理
阿新 • • 發佈:2018-04-10
void log 打表 數學 mod turn ret iostream toc
組合數學推推推最後,推得要求C(n+m,m)%p
其中n,m小於10^9,p小於1^5
用Lucas定理求(Lucas定理求nm較大時的組合數)
因為p數據較小可以直接階乘打表求逆元
求逆元時,由費馬小定理知道p為素數時,a^p-1=1modp可以寫成a*a^p-2=1modp
所以a的逆元就是a^p-2,
可以求組合數C(n,m)%p中除法取模,將其轉化為乘法取模 即 /(m!*(n-m)!)=*(m!*(n-m)!)^p-2
[cpp] view plain copy
- #include <iostream>
- #define ll long long
- const int N=1e5+5;
- using namespace std;
- ll fac[N],p;
- void init()
- {
- fac[0]=1;
- for(int i=1;i<=p;i++)
- fac[i]=fac[i-1]*i%p;
- }
- ll qpow(ll a,ll b)
- {
- ll ans=1;
- a%=p;
- while(b)
- {
- if(b&1)
- {
- ans=ans*a%p;
- }
- b>>=1;
- a=a*a%p;
- }
- return ans;
- }
- ll C(ll a,ll b)
- {
- if(a<b) return 0;
- return fac[a]*qpow(fac[b]*fac[a-b],p-2)%p;
- }
- ll Lucas(ll a,ll b)
- {
- if(b==0) return 1;
- return (C(a%p,b%p)*Lucas(a/p,b/p))%p;
- }
- int main()
- {
- int T;
- ll n,m;
- cin>>T;
- while(T--)
- {
- cin>>n>>m>>p;
- init();
- cout<<Lucas(n+m,m)<<endl;
- }
- return 0;
- }
hdu 3037 費馬小定理+逆元求組合數+Lucas定理