leetcode 70. 爬楼梯（climbing stairs）C语言

1.description

https://leetcode-cn.com/problems/climbing-stairs/description/

假设你正在爬楼梯。需要 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 阶

2.solution

简单经典的DP，C语言实现：

int climbStairs(int n){
    if(n == 1) return 1;
    if(n == 2) return 2;
    int *dp = (int*)malloc(sizeof(int)*(n+1));
    dp[1] = 1;
    dp[2] = 2;
    for(int i=3; i<n+1; ++i){
        dp[i] = dp[i-1] + dp[i-2];
    }
    int res = dp[n];
    free(dp);

    return res;
}

其实用不到数组，每个状态只跟前两个值有关，三个变量就可以了

int climbStairs(int n){
    if(n == 1) return 1;
    if(n == 2) return 2;

    int pre = 1;
    int cur = 2;
    int next;
    for(int i=3; i<n+1; ++i){
        next = pre + cur;
        pre = cur;
        cur = next;
    }
    return cur;
}