計算組合數

編寫函數，參數是兩個非負整數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。因為組合數的公式，分子是大于分母的，所以分子的乘積比較大，容易溢出，要想優(yōu)化，就得讓分子乘積變小。
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; }

汝佳老師提示

“對復雜的表達式進行化簡有時不僅能減少計算量，還能減少甚至避免中間結果的溢出。”

總結

以上是生活随笔為你收集整理的《算法竞赛入门经典》计算组合数问题的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。