1. 程式人生 > >hdu 3037 費馬小定理+逆元求組合數+Lucas定理

hdu 3037 費馬小定理+逆元求組合數+Lucas定理

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
  1. #include <iostream>
  2. #define ll long long
  3. const int N=1e5+5;
  4. using namespace std;
  5. ll fac[N],p;
  6. void init()
  7. {
  8. fac[0]=1;
  9. for(int i=1;i<=p;i++)
  10. fac[i]=fac[i-1]*i%p;
  11. }
  12. ll qpow(ll a,ll b)
  13. {
  14. ll ans=1;
  15. a%=p;
  16. while(b)
  17. {
  18. if(b&1)
  19. {
  20. ans=ans*a%p;
  21. }
  22. b>>=1;
  23. a=a*a%p;
  24. }
  25. return ans;
  26. }
  27. ll C(ll a,ll b)
  28. {
  29. if(a<b) return 0;
  30. return fac[a]*qpow(fac[b]*fac[a-b],p-2)%p;
  31. }
  32. ll Lucas(ll a,ll b)
  33. {
  34. if(b==0) return 1;
  35. return (C(a%p,b%p)*Lucas(a/p,b/p))%p;
  36. }
  37. int main()
  38. {
  39. int T;
  40. ll n,m;
  41. cin>>T;
  42. while(T--)
  43. {
  44. cin>>n>>m>>p;
  45. init();
  46. cout<<Lucas(n+m,m)<<endl;
  47. }
  48. return 0;
  49. }

hdu 3037 費馬小定理+逆元求組合數+Lucas定理