fzu 2020 组合数取模

fzu 2020 组合

题目：

给出组合数C(n,m), 表示从n个元素中选出m个元素的方案数。例如C(5,2) = 10, C(4,2) = 6.可是当n,m比较大的时候，C(n,m)很大！于是xiaobo希望你输出 C(n,m) mod p的值！
Input
输入数据第一行是一个正整数T,表示数据组数 (T <= 100) 接下来是T组数据，每组数据有3个正整数 n, m, p (1 <= m <= n <= 10^9, m <= 10^4, m < p < 10^9, p是素数)
Output
对于每组数据，输出一个正整数，表示C(n,m) mod p的结果。
Sample Input
2
5 2 3
5 2 61
Sample Output
1
10

题意：

给求C(n,m)%p的结果；

思路：

C(n,m)=n*(n-1) (n-2) …… *(n-m+1)/(m!);
因为模运算不兹次除法，所以直接计算师肯定要爆炸的，所以这个时候就要用到计算逆元的方法：
（1）求一个数a的模p（一般为质数，否则有些数求不出逆元）的逆元。
(1)
费马小定理：
费马小定理是数论中的一个重要定理，其内容为： 假如p是质数，且(a,p)=1，那么 a^(p-1) ≡1（mod p）。即：假如a是整数，p是质数，且a,p互质，那么a的(p-1)次方除以p的余数恒等于1。
变形就有：a^(p-2)≡p^(-1)（mod p）;

则a的逆元是（a%p）^(p - 2)，要求p为质数，且（a，p）= 1，a为整数。注意：a>p时先取模。

(2)对于大组合数取模，n,m不大于10^5的话，用逆元的方法，可以解决。对于n,m大于10^5的话，那么要求p<10^5，这样就是Lucas定理了，将n,m转化到10^5以内解。
lucas定理：当p是素数的时候有：

那么对于组合数C(n,m):

所以先用lucas定理分解n,m;
对对于每一组ni，mi。
C(ni,mi)=n*(n-1) * (n-2) …… (n-m+1)/(m!);
=n* (n-1) * (n-2) * …… * (n-m+1) * inv(m!);
=n* (n-1) * (n-2) * …… * (n-m+1) * (m！)^(p-2)
(p为素数，题目给了是质数，所以难度被强行简化了)；

代码：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <cmath>
#include <queue>
using namespace std;
typedef long long ll;
ll n,m,q;
ll quick_mod(int x,int y)
{
    ll ans=1;
    ll res=x;
    while(y!=0)
    {
        if(y%2==1)
            ans=ans*res%q;
        res=res*res%q;
        y>>=1;
    }
    return ans;
}
ll C(int x,int y)
{
    ll ans=1;
    for(int i=1;i<=y;i++)
    {
        ll a=(x-y+i);
        ll b=i%q;
        ans=((ans*a%q)*quick_mod(b,q-2))%q;
    }
    return ans;
}
ll lucas(ll x,ll y)
{
    if(y==0)
        return 1;
    else
        return (C(x%q,y%q)*lucas(x/q,y/q))%q;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld%lld%lld",&n,&m,&q);
        ll ans=lucas(n,m);
        printf("%lld\n",ans);
    }
    return 0;
}

感觉应该就是这个意思。