LintCode | 111. 爬楼梯

假设你正在爬楼梯，需要n步你才能到达顶部。但每次你只能爬一步或者两步，你能有多少种不同的方法爬到楼顶部？
题目链接

DP基础题，不多说了。取模可以减少空间浪费。

public class Solution {
    /**
     * @param n: An integer
     * @return: An integer
     */
    public int climbStairs(int n) {
        if(n == 0) return 1;
        int[] dp = new int[3];
        dp[0] = 1;
        dp[1] = 2;
        for(int i = 2; i < n; i++) {
            dp[i % 3] = dp[(i - 1) % 3] + dp[(i - 2) % 3];
        }
        return dp[(n-1) % 3];
    }
}

习惯了DP的标准写法，网上其他人的DP写法有不用数组的，代码如下，喜欢哪种见仁见智了。

public class Solution {
    /**
     * @param n: An integer
     * @return: An integer
     */
    public int climbStairs(int n) {
        int result = 1;
        int first = 1;
        int next = 1;
        while (--n > 0)
        {
            result = first + next;
            first = next;
            next = result;
        }

        return result;
    }
}