直接切入正题:
给一个组合数C(n,k),用公式直接求固然可以,但是直接求阶乘会有两个麻烦:
1、数据过大时,k!和(n!/(n-k)!) (尽管已经化简)还是很容易溢出int的范围或longlong的范围,我们当然不能用mod运算来规避,因为mod运算是不能用在除法上的,即(a/b) mod p != (a mod p) / (b mod p)。
2、我们总共需要计算k!+(n!/(n-k)!) 次乘法+一次除法,这是一个O(n!)的算法,很容易超时。
动态规划解法:
初中有一种类型的题,可以将矩阵和组合数相转化:
给一个5*3的矩阵a,一个人在最左上端,即a[0][0],他只能往下走和往右走,问他到a[4][2](即最右下端)有多少种走法。
问题分析:
步数已经确定了,即向下走4步,向右走两步,一共是6步,所以我们只用在6步之中选两步让他向右走即可,其余一直向下即可,即总方法数为C(6,2)。
如何从组合数转化为矩阵呢?
给一个组合数C(n,k),即在n步之中选k步让他向右走,其余的(n-k)步要让他向下走(也可以n-k步向右,n步向下,对结果不影响),所以我们只需构造一个[n-k+1][k+1](或[k+1][n-k+1])的矩阵,便完成了问题的转化(以此问题作为桥梁)。
在矩阵下,用动态规划(也就是递推)的方法对问题进行求解,从小问题的解得到大问题的解。
假如矩阵的大小是(1,1),那么这个人直接不用走了,即方法数为1(不走也算一种方法吧hhhh),d[0][0]=1;
矩阵的大小是(1,z)(z>=2),那么这个人只需一直向右走即可,方法数为1,d[0,z-1]=1, 即d[0,j]=1;
同理也知道矩阵的大小是(z,1)(z>=2)时的方法数也为1,d[z-1,0]=1,即d[i,0]=1;
那么更普遍的(z,z)(矩阵大小为(z,z)时)呢?
d[z-1,z-1]=此人走到左边格子的方法数+此人走到上边格子的方法数=d[z-2][z-1]+d[z-1][z-2];
即d[i][j]=d[i-1][j]+d[i][j-1] (i,j != 0)
公式推导完毕。
C++代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int Cnk(int n,int k) //C(n,k)
{
int d[k+1][n-k+1]; //声明一个n-1行n-1列的矩阵
memset(d,0,sizeof(d));
for(int i=0;i<n-k+1;i++) //初始化第一行
{
d[0][i]=1;
}
for(int i=1;i<k+1;i++) //初始化第一列
{
d[i][0]=1;
}
for(int i=1;i<k+1;i++)
{
for(int j=1;j<n-k+1;j++)
{
d[i][j]=d[i-1][j]+d[i][j-1];
}
}
for(int i=0;i<k+1;i++)
{
for(int j=0;j<n-k+1;j++)
{
printf("%3d ",d[i][j]);
}
printf("\n");
}
return d[k][n-k];
}
int main()
{
int n,k;
printf("请输入组合数的n和k\n");
cin>>n>>k;
printf("组合数C(%d,%d)的结果是%d\n",n,k,Cnk(n,k));
return 0;
}