leetcode70. 爬楼梯 动态规划问题

假设你正在爬楼梯。需要 n 阶你才能到达楼顶。

每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？

注意：给定 n 是一个正整数。

示例 1：

输入： 2
输出： 2
解释： 有两种方法可以爬到楼顶。
1.  1 阶 + 1 阶
2.  2 阶

示例 2：

输入： 3
输出： 3
解释： 有三种方法可以爬到楼顶。
1.  1 阶 + 1 阶 + 1 阶
2.  1 阶 + 2 阶
3.  2 阶 + 1 阶

方法1：记忆化搜索

#include<iostream>
#include<map>

using namespace std;

// 记忆话搜索
class Solution {
public:
	int climbStairs(int n) {
		if (n == 0 || n == 1)
			return 1;
		if (memo.count(n) == 0) // memo中是否有值
			memo[n] = climbStairs(n - 1) + climbStairs(n - 2);

		return memo[n];
	}

private:
	map<int, int> memo;
};

int main()
{
	int n;
	cin >> n;
	cout << Solution().climbStairs(n) << endl;

	return 0;
}

方法2：动态规划

#include<iostream>
#include<vector>

using namespace std;

class Solution {
public:
	int climbStairs(int n) {
		vector<int> memo(n + 1, -1);
		memo[0] = 1;
		memo[1] = 1;
		for (int i = 2; i <= n; i++)
			memo[i] = memo[i - 1] + memo[i - 2];

		return memo[n];
	}
};

int main()
{
	int n;
	cin >> n;
	cout << Solution().climbStairs(n) << endl;

	return 0;
}