最近写题已经遇到过两次要用组合数C(n,k)的题目了,为了方便下次用板子,写个博客记录一下
方法一:在数据比较小的时候,可以用先乘后除的方法
以下代码复杂度为O(n+k) 缺点是容易爆long long
我们知道,C(n,m)=[ n*n-1*n-2* .... *(n-k) ] / [1*2*3* .... *k],下方代码便是
long long C(long long nn, long long kk)
{
long long ansC = 1;
for (int i = nn; i > nn - kk; i--) ansC *= i;
for (int i = kk; i > 1; i--) ansC /= i;
return ansC;
}
方法二:利用递推公式:C(n,k) = (n-k+1)/k*C(n,k-1),递推求组合数,复杂度为O(k)
代码如下:
long long C(long long nn, long long kk)
{
ll CC[1010]={0};
CC[0]=1;
for(int i = 1; i <= kk; i++)
CC[i] = CC[i-1]*(nn-i+1)/i;
//返回C(n,k)
return CC[kk];
}
据紫书上得:递推得时候应该是先乘再除,因为如果先除的话C[i-1]/i可能不是一个整数,但如果先乘的话,可能会爆long long。(ps:实在不行可以先约分)
方法三:利用递推公式:C(n,k) = (n-k+1)/k*C(n,k-1),递推求组合数,代码如下:
long long C(long long nn,long long kk)
{
long long CC[1010][1010]={0};
for(int i = 0; i <= nn; i++)
{
CC[i][0] = 1;
for(int j = 1; j<=i; j++)
CC[i][j] = CC[i-1][j-1] + CC[i-1][j];
}
return CC[nn][kk];
}
const int maxn=1010;
long long C[maxn][maxn];
void CC(long long nn)//直接造一个组合数数组
{
memset(C,0,sizeof C);//初始化 调用此函数需要#include<cstring>
for(int i = 0; i <= nn; i++)
{
C[i][0] = 1;
for(int j = 1; j<=i; j++)
C[i][j] = C[i-1][j-1] + C[i-1][j];
}
}