递归climbing-stairs-leetcode练习题

刺骨的言语ヽ痛彻心扉 2023-07-19 05:54 10阅读 0赞

import java.io.BufferedReader;
    import java.io.IOException;
    import java.io.InputStreamReader;
    import java.util.HashMap;
    import java.util.Map;
    
    /**
     * 假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
     * <p>
     * 每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？
     * <p>
     * 注意：给定 n 是一个正整数。
     * <p>
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/climbing-stairs
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     */
    public class Solution {
        /**
         * 缓存
         */
        Map<Integer, Integer> cache = new HashMap<>();
    
        /**
         * @param n
         * @return
         */
        public int climbStairs(int n) {
            if (n == 1) {
                return 1;
            }
            if (n == 2) {
                return 2;
            }
            int res1 = 0;
            int res2 = 0;
            if (cache.get(n - 1) == null) {
                res1 = climbStairs(n - 1);
                cache.put(n - 1, res1);
            }else{
                res1 =cache.get(n-1);
            }
            if (cache.get(n - 2) == null) {
                res2 = climbStairs(n - 2);
                cache.put(n - 2, res1);
            }else{
                res2 = cache.get(n-2);
            }
            return res1 + res2;
        }
    
        public static void main(String[] args) throws IOException {
            BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
            String line;
            while ((line = in.readLine()) != null) {
                int n = Integer.parseInt(line);
                int ret = new Solution().climbStairs(n);
                String out = String.valueOf(ret);
                System.out.print(out);
            }
        }
    }

解题思路

递归+缓存