组合数的一些模板(好用)

news/2024/8/26 16:20:19 标签: 模板, 组合数

一、\LARGE {\color{Red} a^{b}\, \, mod\, \, p}

ll qpow(ll a,ll b,ll p)
{
	ll ret=1;a%=p;
	while(b)
	{
		if(b&1) ret=ret*a%p;
		b/=2;a=a*a%p;
	}
	return ret;
}

二、Lucas

ll lucas(ll n,ll m,ll p)
{
	if(m==0) return 1;
	return C(n%p,m%p,p)*lucas(n/p,m/p,p)%p;
}

三、C(n,m)

方案一:(直接算)

 

ll C(ll n,ll m,ll p)
{
	if(m>n) return 0;
	ll ans=1;
	for(int i=1;i<=m;i++)
	  {
	  	ll a=(n+i-m)%p;
	  	ll b=i%p;
	  	ans=ans*(a*qpow(b,p-2,p)%p)%p;
	  }
	return ans;
}

方案二:(预处理)

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=b;i>=a;i--)
using namespace std;
#define ll long long
const int N=2010;
const int mod = 998244353;
int fac[N+10],inv[N+10]; 
ll rd()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int sum(int a, int b) {
    int s = (a + b);
    if (s >= mod) s -= mod;
    return s;
}
int sub(int a, int b) {
    int s = a - b;
    if (s < 0) s += mod;
    return s;
}
int mult(int a, int b) {
    return (1LL * a * b) % mod;
}
ll qpow(ll a,ll b)
{
	ll ret=1;a%=mod;
	while(b)
	{
		if(b&1) ret=ret*a%mod;
		b/=2;a=a*a%mod;
	}
	return ret;
}
int C(int m,int n)
{
	if(m>n) return 0;
	return 1ll * fac[n] * inv[m] % mod * inv[n - m] % mod;
}
int main()
{
	fac[0]=inv[0]=1;
	rep(i,1,N) fac[i]=mult(fac[i-1],i);
	inv[N]=qpow(fac[N],mod-2);
	dep(i,1,N-1) inv[i]=mult(inv[i+1],i+1);
} 
ll C(ll n,ll m)
{
	if(m>n) return 0;
	return 1ll * fac[n] * inv[m] % p * inv[n - m] % p;
}
   
fac[0]=1;inv[0]=1;
for(int i=1;i<=p;i++) 
{
    fac[i]=1ll*fac[i-1]*i%p;
	inv[i]=qpow(fac[i],p-2);	
}

fac[0]=inv[0]=1;
rep(i,1,100) fac[i]=mult(fac[i-1],i);
inv[100]=qpow(fac[100],mod-2);
dep(i,1,99) inv[i]=mult(inv[i+1],i+1);

方案三:(预处理质数)

void isprime()
{
	for(int i=2;i<=N;i++)
	{
		if(!flag[i]) prime[++l]=i;
		num[i]=l;
		for(int j=1;j<=l;j++)
		{
			if(1ll*prime[j]*i>N) break;
			flag[prime[j]*i]=1;
			if(i%prime[j]==0) break;
		}
	}
}
void init()
{
	for(int i=1;i<=l;i++)
	{
		//printf("%d\n",i);
		f[i][0]=inv[i][0]=1;
		for(int j=1;j<prime[i];j++)
		{
			f[i][j]=(f[i][j-1]*j)%prime[i];
			inv[i][j]=qpow(f[i][j],prime[i]-2,prime[i]);
		}
	}
}
ll C(ll n,ll m,ll p)
{
	if(m>n) return 0;
	if(m==n) return 1;
	int t=num[p];//第几个质数
	return f[t][n]*(inv[t][n-m]*inv[t][m]%p)%p;
}

 

四、组合数+中国剩余定理

ll gcd(ll a,ll b)
{
    return b==0?a:gcd(b,a%b);
}
void exgcd(ll a,ll b,ll &x,ll &y)
{
    if(b==0){x=1;y=0;return;}
    exgcd(b,a%b,x,y);
    ll t=x;
    x=y;y=t-a/b*y;
}
ll work(int n)
{
	ll x,y,a,b,c,t,U=u[1],V=v[1];
	rep(i,2,n)
	{
	  a=U;b=u[i];c=v[i]-V;
	  t=gcd(a,b);
	  if(c%t!=0) return -1;
	  a/=t;b/=t;c/=t;
	  exgcd(a,b,x,y);
	  x=((c*x)%b+b)%b;
	  if(!x) x+=b;
	  V=U*x+V,U=a*b*t,V%=U;
	}
	return (V%U+U)%U;
}
ll qpow(ll a,ll b,ll p)
{
	ll ans=1;
	a%=p;
	while(b)
	  {
	  	if(b&1) {ans=ans*a%p;b--;}
	  	b>>=1;a=a*a%p;
	  }
	return ans;
}
ll C(ll n,ll m,ll p)
{
	if(m>n) return 0;
	ll ans=1;
	for(int i=1;i<=m;i++)
	  {
	  	ll a=(n+i-m)%p;
	  	ll b=i%p;
	  	ans=ans*(a*qpow(b,p-2,p)%p)%p;
	  }
	return ans;
}
ll Lucas(ll n,ll m,ll p)
{
	if(m==0) return 1;
	return C(n%p,m%p,p)*Lucas(n/p,m/p,p)%p;
}

 


http://www.niftyadmin.cn/n/943252.html

相关文章

bzoj 1934: [Shoi2007]Vote 善意的投票

Description 幼儿园里有n个小朋友打算通过投票来决定睡不睡午觉。对他们来说&#xff0c;这个问题并不是很重要&#xff0c;于是他们决定发扬谦让精神。虽然每个人都有自己的主见&#xff0c;但是为了照顾一下自己朋友的想法&#xff0c;他们也可以投和自己本来意愿相反的票。我…

bzoj 1706: [usaco2007 Nov]relays 奶牛接力跑

Description FJ的N(2 < N < 1,000,000)头奶牛选择了接力跑作为她们的日常锻炼项目。至于进行接力跑的地点 自然是在牧场中现有的T(2 < T < 100)条跑道上。 农场上的跑道有一些交汇点&#xff0c;每条跑道都连结了两个不同的交汇点 I1_i和I2_i(1 < I1_i < 1,0…

杭电ACM——1247,Hat’s Words(Trie树)

这道题是有问题的&#xff0c;一个单词是可以被另外一个单词拼接两次得到的&#xff0c;如ab&#xff0c;abab&#xff0c;输出是abab&#xff0c;然而原题确指明&#xff0c;某个单词当且仅当是另外字典里的两个单词拼接的&#xff0c;才是一个hat’s word&#xff0c;实际上这…

bzoj 1001: [BeiJing2006]狼抓兔子

Description 现在小朋友们最喜欢的"喜羊羊与灰太狼",话说灰太狼抓羊不到&#xff0c;但抓兔子还是比较在行的&#xff0c;而且现在的兔子还比较笨&#xff0c;它们只有两个窝&#xff0c;现在你做为狼王&#xff0c;面对下面这样一个网格的地形&#xff1a; 左上角点…

杭电ACM——2533,N皇后问题(搜索)

突破口&#xff1a; N*N的棋盘放置N个皇后&#xff0c;因为皇后的特性&#xff0c;合法的放置方法中每行&#xff08;每列&#xff09;必定有一个皇后&#xff0c;因此我们可以先搜索第1行可以放置的位置&#xff0c;再逐步搜索第2行&#xff0c;直到第N行。而不用去穷举第一个…

杭电ACM——2546,饭卡(DP)

一维背包问题。 突破口&#xff1a;将菜的价格从小到大排序&#xff0c;对前n-1个物品进行DP&#xff0c;找出消费金额不超过m-5的方案中最大的消费金额sum&#xff0c;在此基础上&#xff0c;m - (sumprice[n]) 即为卡内最小余额。 因此&#xff0c;我们DP应在余额为0~m-5之内…

bzoj 1797: [Ahoi2009]Mincut 最小割

Description A,B两个国家正在交战&#xff0c;其中A国的物资运输网中有N个中转站&#xff0c;M条单向道路。设其中第i (1≤i≤M)条道路连接了vi,ui两个中转站&#xff0c;那么中转站vi可以通过该道路到达ui中转站&#xff0c;如果切断这条道路&#xff0c;需要代价ci。现在B国想…

北大ACM——3046,Ant Counting(DP)

题意&#xff1a;有T种蚂蚁&#xff0c;总共A只&#xff0c;求将这些蚂蚁&#xff0c;分为数量为S,S1,S2,…,B的集合的总集合数。 突破口&#xff1a; 设置a[T]记录每种蚂蚁的数量&#xff0c;设置dp[i,j]表示&#xff0c;用前i种蚂蚁&#xff0c;构成数量为j的集合的最大集合数…