组合(Lucas)

Lucas 是用来求组合数取模的。
Lucas 定理:A、B是非负整数,p是质数。AB写成p进制:A=a[n]a[n-1]…a[0],B=b[n]b[n-1]…b[0]。
可以求逆元来解决,用费马小定理,或者是扩展欧几里得。
给出组合数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

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
long long int n, m, p;
long long int quick_mod(long long int a, long long int b)
{
    long long int ans = 1;
    a = a%p;
    while(b)
    {
        if(b&1)
            ans = ans * a % p;
        b>>=1;
        a = a * a % p;
    }
    return ans;
}
long long int C(long long int n, long long int m)
{
    if(m>n)
        return 0;
    long long int ans = 1;
    for(int i=1;i<=m;i++)
    {
        long long int a= (n+i-m)%p;
        long long int b = i%p;
        ans = ans *(a * quick_mod(b, p-2)%p)%p;
    }
    return ans;
}
long long int Lucas(long long int n, long long int m)
{
    if(m==0)
        return 1;
    return C(n%p, m%p)*Lucas(n/p, m/p)%p;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        scanf("%lld %lld %d", &n, &m, &p);
        cout<<Lucas(n, m)<<endl;
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值