// 题目:
// 假设你正在爬楼梯,需要n步你才能到达顶部。但每次你只能爬一步或者两步,你能有多少种不同的方法爬到楼顶部?
#include <iostream>
using namespace std;
// 思路:
// 利用DP的方法,一个台阶的方法次数为1次,两个台阶的方法次数为2个。n个台阶的方法可以理解成上n-2个台阶,然后2步直接上最后一步;或者上n-1个台阶,再单独上一步。
// 公式是ways[n] = ways[n-1] + ways[n-2] , ways[1] = 1, ways[2] = 2
// 每次至多只能爬两步的情形
int ClimbStairs2(int n) {
if (n <= 2) return n;
int ways = 0;
int step1 = 1, step2 = 2;
for (int i = 2; i < n; i++) {
ways = step1 + step2;
step1 = step2;
step2 = ways;
}
return ways;
}
// 扩展
// 每次至多只能爬三步的情形
int ClimbStairs3(int n) {
if (n <= 2) return n;
if (n == 3) return 4;
int ways = 0;
int step1 = 1, step2 = 2, step3 = 4;
for (int i = 3; i < n; i++) {
ways = step1 + step2 + step3;
step1 = step2;
step2 = step3;
step3 = ways;
}
return ways;
}
int main() {
int n;
while (cin >> n) {
cout << ClimbStairs3(n) << endl;
}
return 0;
}
Leetcode | Climbing Stairs
最新推荐文章于 2021-08-21 16:36:22 发布