1. 程式人生 > >LeetCode-70. 爬樓梯

LeetCode-70. 爬樓梯

假設你正在爬樓梯。需要 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 階

這題要用一下記憶化搜尋然而LeetCode的測試資料比較嚴苛不知道有陣列開多少又不想試所以我用一個hashmap去充當陣列

solution:

class Solution {
    
    static HashMap<Integer,Integer> hashMap = new HashMap<>();
    
    public int climbStairs(int n) {
        if(hashMap.containsKey(n))
            return hashMap.get(n);
        else{
            if(n==1){
                hashMap.put(1,1);
                return 1;
            }
            else if(n==2){
                hashMap.put(2,2);
                return 2;
            }
            else{
                int val = climbStairs(n-1)+climbStairs(n-2);
                hashMap.put(n,val);
                return val;
            }
        }
    }
}