计算组合数
编写函数,参数是两个非负整数n和m,返回组合数
其中m<=n<=25。例如,n=25,m=12时的答案为5200300。
代码及算法分析
程序4-1 组合数(有问题)
#include <stdio.h>
long long factorial(int n)
{
long long m=1;
//不要忘记初始化,不然的出来的结果惊人。
for(int i=1;i<=n;i++)
{
m*=i;
}
return m;
}
int main ()
{
int n,m;
scanf("%d %d",&n,&m);
printf("%lld",factorial(n)/(factorial(m)*factorial(n-m)));
return 0;
}
这个代码的问题显而易见,阶乘容易溢出,所以不可取。
所以,套用刘汝佳老师的一句话:”即使最终答案在所选择的数据类型范围之内,计算的中间结果仍然可能溢出“。
汝佳老师也给出了相应的解决方案,虽然不能完全避免中间结果溢出,但是对于题目给出的范围已经可以保证得到正确的结果了。
先来分析一下组合数的公式:
n!
————
m!(n-m)!
展开之后:
n *(n-1) *(n-2)……3 *2 *1
—————————————————————————————(1)
m *(m-1) *(m-2)……3 *2 *1 *(n-m) *(n-m-1) *(n-m-2)…… *3 *2 *1
因为n>=m,所以m在1~n之间,这样可以把n!除以m!约分:
n *(n-1) *(n-2)……(m+2) *(m+1)
————————————————(2)
(n-m) *(n-m-1) *(n-m-2)…… *3 *2 *1
然后汝佳老师给了一道思考题:为什么当m<n-m时要把m变成n-m?
移项之后得到:m<n/2,先列一个数轴:
|————————————————————>
0 1 2 3 ……m……n/2……M……(n-1) n
为了不把变换后的m与变换前的m弄混,把变化后的m记为M。因为组合数的公式,分子是大于分母的,所以分子的乘积比较大,容易溢出,要想优化,就得让分子乘积变小。
M=n-m带入组合式公式之后就是:
n!
————————————————(3)
M!(n-M)!=(n-m)!(n-(n-m))!=(n-m)!m!
所以式子并没有变化,同样可以进行约分得到(2)式的变式:
n *(n-1) *(n-2)……(M+2) *(M+1)
————————————————(4)
(n-M) *(n-M-1) *(n-M-2)…… *3 *2 *1
相当于将数轴
|————————————————————>
0 1 2 3 ……m ……n/2……M ……(n-1) n
标出的部分直接干掉了。
由于M是大于m的,所以分子的乘积缩小,溢出问题得到缓解,注意,是缓解不是解决,如果n和m的值很大的话还是会溢出。
分析之后就可以把程序4-1升级为程序4-2了:
#include <stdio.h>
long long c(int n,int m)
{
if(m<n-m)
{
m=n-m;
}
long long ans=1;
for(int i=m+1;i<=n;i++)
{
ans*=i;
}
for(int i=1;i<=n-m;i++)
{
ans/=i;
}
return ans;
}
int main ()
{
int n,m;
scanf("%d %d",&n,&m);
printf("%lld",c(n,m));
return 0;
}
汝佳老师提示
“对复杂的表达式进行化简有时不仅能减少计算量,还能减少甚至避免中间结果的溢出。”