【题目描述】
小明爬楼梯,一次可以爬一个或两个台阶;如果楼梯有20个台阶,请问小明一共有多少种爬法?
【解决思路】
到每一个台阶,可以由前一个台阶上来,或者由前两个台阶上来,该思路符合斐波那契数列的规律:F(n) = \F(n-1) + \F(n-2)
【代码演示】
//一次可以走1个或2个台阶
public static int getSum2(int step){
if (step==1){
return 1;
}else if (step==2){
return 2;
}
return getSum2(step-1)+getSum2(step-2);
}
//一次可以走1个、2个、或3个台阶
public static int getSum3(int step){
if (step==1){
return 1;
}else if (step==2){
return 2;
}else if (step==3){
return 4;
}
return getSum3(step-1)+getSum3(step-2)+getSum3(step-3);
}
//一次可以走1~4个台阶
public static int getSum4(int step){
if (step==1){
return 1;
}else if (step==2){
return 2;
}else if (step==3){
return 4;
}else if (step==4){
return 8;
}
return getSum4(step-1)+getSum4(step-2)+getSum4(step-3)+getSum4(step-4);
}
由上可以改进代码为:
//一次可以走1~n个台阶,一共step级台阶
public static int getSumN(int n, int step) {
int[] arr = new int[n];
arr[0] = 1;
int sum = 0;
//前n个台阶的爬法数=前n-1个台阶爬法累加之和 + 1(1代表一步到位的走法:0级到n级台阶)
for (int i = 1; i < n; i++) {
sum = sum + arr[i - 1];
arr[i] = sum + 1;
}
int[] arrs = new int[step >= n ? step : n];
System.arraycopy(arr, 0, arrs, 0, arr.length);
for (int i = arr.length; i < step; i++) {
int c = 0;
//第i级台阶的到法 = 往前推的各种走法累加之和 (和上面return的规律一样)
for (int j = 1; j <= n; j++) {
c = c + arrs[i - j];
}
arrs[i] = c;
}
return arrs[step - 1];
}
}