【fzu 2020】 组合

Problem 2020 组合

Accept: 1331    Submit: 3196
Time Limit: 1000 mSec    Memory Limit : 32768 KB

Problem Description

给出组合数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

25 2 35 2 61

Sample Output

110

Source

FOJ有奖月赛-2011年04月（校赛热身赛）

题解1：C(n,m)=C(n,m-1)*(n-m+1)/m

#include<stdio.h>
#include<iostream>
#define ll long long
using namespace std;
ll exgcd(ll a,ll b,ll &x,ll &y){
    if(!b){
        x=1,y=0;
        return a;
    }
    ll g=exgcd(b,a%b,y,x);
    y-=a/b*x;
    return g;
}
ll inv(ll x,ll p){
    ll a,b;
    exgcd(x,p,a,b);
    return (a+p)%p;
}
void read(ll &s){
	s=0;
	char c=getchar();
	while(c<'0'||c>'9'){
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		s*=10;
		s+=c-'0';
		c=getchar();
	}
}
int main(){
	int t,i;
	ll n,m,p,ans;
	scanf("%d",&t);
	while(t--){
		read(n);
		read(m);
		read(p);
		ans=n%p;
		for(i=2;i<=m;i++){
			ans*=inv(i,p);
			ans%=p;
			ans*=n-i+1;
			ans%=p;
		}
		printf("%lld\n",ans);
	}
	return 0;
}

题解2：

相对来说最简单的一种——扩展欧几里得求逆元。刚拿到题目时我想的是直接用lucas定理的模板来做。可是我们观察数据范围，可以发现n和p非常大，但m非常小。再考虑到组合数的公式：C(n,m)=n!/m!(n-m)!。如果直接把n!/(n-m)!合并以后再除以m!，就可以做到分式上下都只有m个数，于是直接枚举n-m+1到n，每个数累乘起来模mod，再枚举1到m，每个数在模mod下的逆元累乘起来模mod，于是我们就得到了最终的答案！

#include<iostream>
#include<cstdio>
#define ll long long
using namespace std;
int t;
ll exgcd(ll a,ll b,ll &x,ll &y){  
    ll d;  
    if(b){d=exgcd(b,a%b,y,x);y-=a/b*x;return d;}  
    x=1;y=0;return a;  
}  
ll inv(ll x,ll mod){
    ll anx,any;exgcd(x,mod,anx,any);return (anx+mod)%mod;  
}  
void work(){
	ll n,m,p,ans1=1,ans2=1;
	cin>>n>>m>>p;
	for(ll i=n-m+1;i<=n;i++){
		ans1=ans1*i;ans1=ans1%p;
	}
	for(ll i=2;i<=m;i++){
		ans2=ans2*i;ans2=ans2%p;
	}
	printf("%lld\n",ans1*inv(ans2,p)%p);
}
int main(){
	cin>>t;
	while(t--){
		work();
	}
	return 0;
}

题解3：Lucas

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
long long mexp(long long a,long long b,long long c){
	long long s=1;
	while(b){
		if(b&1)s=s*a%c;
		a=a*a%c;
		b>>=1;
	}
	return s;
}
long long exgcd(long long a,long long b,long long &x,long long &y){
	if(!b){
		x=1;y=0;
		return a;
	}
	long long s=exgcd(b,a%b,x,y),t;
	t=x;x=y;y=t-a/b*y;
	return s;
}
long long inv(long long a,long long b){
	long long x,y;
	long long d=exgcd(a,b,x,y);
	return (x%b+b)%b;
}
long long lucas(int n,int m,long long p){
	if(n<m)return 0;
	if(m>=p||n>=p)return lucas(n/p,m/p,p)%p*lucas(n%p,m%p,p)%p;
	if(m>n-m)m=n-m;
	long long s1=1,s2=1;
	for(int i=n-m+1;i<=n;i++)s1=s1*i%p;
	for(int i=1;i<=m;i++)s2=s2*i%p;
	return s1*inv(s2,p)%p;
}
int main(){
	int t=read(),n,m,p;
	while(t--){
		n=read();m=read();p=read();
		printf("%lld\n",lucas(n,m,p));
	}
	return 0;
}