每周一道脱发题，如何优雅的爬楼梯

问题描述：

楼梯有n阶台阶，上楼可以一步上1阶，也可以一步上2阶，计算共有多少种不同的走法？

 

分析：

如果只有1个台阶，走法有1种（一步上1个台阶），即f(1)=1；如果有2个台阶，走法有2种（一种是上1阶，再上1阶，另一种是一步上2阶），即f(2)=2；如果有3个台阶，走法有4种（一种每次1阶，共一种；另一种是2+1，共两种；第三种是3，共1种），即f(3)=4；

我们可以看到3个台阶，其实可以认为是f(1)，再走2阶，或者f(2),再走一个台阶。同理f(4)=走完f(3)再走1阶+走完f(2)再走2阶。也就是我们可以得出当n>2时公式,f(n)=f(n-2)+f(n-1)

很容易想到我们可以自己定义出口（爬一阶楼梯和爬两阶楼梯的走法数量）完成一个递归的算法：

public int climbStairs(int n) {
	if (n <= 0) {
		return 0;
	}
	if (n == 1 || n == 2) {
		return n;
	}

	// f(n)=f(n-2)+f(n-1)
	return climbStairs(n - 1) + climbStairs(n - 2);
}

精进：

这个算法在实际计算一个非常大的数字的时候就非常慢了。原因是重复计算很多，导致耗时很长，所以需要对这个程序进行优化。既然涉及到重复计算，那么我们很容易想到保存中间结果，如果已经计算过了，那么直接返回中间结果值，如果没有，再进行递归计算，并把结果保存起来。

List<Integer> resList = new ArrayList<>();

public int climbStairs2(int n) {
	if (n <= 0) {
		return 0;
	}
	if (n == 1 || n == 2) {
		return n;
	}

	if (resList.size() > (n - 2)) {
		return resList.get(n - 3);
	}
	// f(n)=f(n-2)+f(n-1)
	int res = climbStairs2(n - 1) + climbStairs2(n - 2);
	resList.add(res);// 分别装入[f[3],f[4],f[5]......]
	return res;

}

再精进：

这样在实际运行的时候就可以在速度上有一个非常大的提升了。当然你也可以将递归转为循环，减少栈空间使用。

public int climbStairs3(int n) {
	if (n <= 0) {
		return 0;
	}
	List<Integer> res = new ArrayList<>();
	res.add(1);
	res.add(2);

	for (int i = 2; i < n; i++) {
		int sum = res.get(i - 2) + res.get(i - 1);
		res.add(sum);

	}
	return res.get(n - 1);
}

举一反三：

好啦，今天头发就掉到这里吧。不不不，再给大家一个思考题，如果题目改成这样子，又怎样呢？

楼梯有n阶台阶，上楼可以一步上1阶，也可以一步上2阶，也可以一步上3阶，计算共有多少种不同的走法？