昨天说到这个问题,想了一些思路,今天就把它实现了一下,也是收获满满!
问题描述:
这一类的问题估计大家见过很多,大致是这样的:
有n阶台阶,小明每次只能上1个或2个台阶,问他有多少种方案走完n阶台阶。
解决方案:
1、 作为一个标准的程序员,我的第一反应就是:最后一次迈两个台阶or一个台阶,那么问题就表示为:f(n)=f(n-1)+f(n-2)
于是乎,一个递归程序就这么出来了!
//递归实现,f(n)=f(n-1)+f(n-2),时间复杂度O(n*n)
int func1(int n){
if(n==1)
return 1;
if(n==2)
return 2;
return func1(n-1)+func1(n-2);
}
2、学通信的老公第一反应却是,这是一个组合数学问题啊!(果然通信的数学能力要好啊!)
思路描述:
只有两种选择,2或1;假设选a个1,b个2,则a+2b=n;则总共有a+b个坑,从中选择b个坑,这b个坑放两个台阶,剩下的a个坑里面放1个台阶,即C(a+b,b)=C(n-b,b)。b的取值范围为:0~n/2,无论n为奇数或偶数都是一样的。
//组合公式求解,时间复杂度(1+1+2+3+...+n/2=n^2/8+n/4+1=O(n*n))
int func3(int n){
int res = 0;
for(int i=0;i<=n/2;i++){
int fenzi = 1;
int fenmu = 1;
for(int j=0;j<i;j++){
fenzi = (n-i-j)*fenzi;
fenmu = (i-j)*fenmu;
}
res += fenzi/fenmu;
}
return res;
}
3、 时间复杂度有些高,递归重复做了好多工作!
用迭代实现,只需要O(n)的时间复杂度哦!
//迭代实现,f(n)=f(n-1)+f(n-2),时间复杂度O(n)
int func2(int n){
if(n==1)
return 1;
if(n==2)
return 2;
int s1=1,s2=2,s3=0;
for(int i=3;i<=n;i++){
s3 = s1+s2;
s1 = s2;
s2 = s3;
}
return s3;
}
整合代码测试:
分析时间复杂度:迭代优于组合,组合优于递归。
测试证明:
测试n<=23时,三种方案均可行;n>23时,组合的方案会出错,因为数据过大,超出了计算机的存储范围。
综合代码:
#include <iostream>
#include <ctime> //time_t类型所需头文件
using namespace std;
//递归实现,f(n)=f(n-1)+f(n-2),时间复杂度O(n*n)
int func1(int n){
if(n==1)
return 1;
if(n==2)
return 2;
return func1(n-1)+func1(n-2);
}
//迭代实现,f(n)=f(n-1)+f(n-2),时间复杂度O(n)
int func2(int n){
if(n==1)
return 1;
if(n==2)
return 2;
int s1=1,s2=2,s3=0;
for(int i=3;i<=n;i++){
s3 = s1+s2;
s1 = s2;
s2 = s3;
}
return s3;
}
//组合公式求解,时间复杂度(1+1+2+3+...+n/2=n^2/8+n/4+1=O(n*n))
int func3(int n){
int res = 0;
for(int i=0;i<=n/2;i++){
int fenzi = 1;
int fenmu = 1;
for(int j=0;j<i;j++){
fenzi = (n-i-j)*fenzi;
//cout << "fenzi:" << fenzi << endl;
fenmu = (i-j)*fenmu;
//cout << "fenmu:" << fenmu << endl;
}
res += fenzi/fenmu;
//cout << "res:" << res << endl;
}
return res;
}
int main()
{
clock_t start,finish;
int n; //总的台阶数
cin >> n;
start = clock();
cout << func1(n) << endl;
finish = clock();
cout << finish-start << " (ms) "<< endl;
start = clock();
cout << func2(n) << endl;
finish = clock();
cout << finish-start << " (ms) "<< endl;
start = clock();
cout << func3(n) << endl;
finish = clock();
cout << finish-start << " (ms) "<< endl;
return 0;
}