leetcode213. 打家劫舍 II

题目描述

递归 + 缓存

解题思路

代码演示

/**
* 主方法调用
*/
public int rob(int[] nums) {
		
        if(nums.length == 1){
            return nums[0];
        }
        //两个或者三个房间时，因为环形只能选一个，所以选最大值就行了
          if(nums.length <= 3 ){
            Arrays.sort(nums);
            return nums[nums.length - 1];
        }
		//缓存表，第一个房间不选时的缓存表
        int[]ans =  new int[nums.length + 1];
        Arrays.fill(ans,-1);
        //缓存表，第一个房间选时的缓存表
        int[]ans2 =  new int[nums.length + 1];
        Arrays.fill(ans2,-1);
        //第一个房间不选时，最后一个房间可以选，
        //所以结束位置nums.length - 1 ，起始位置1 
        int p1 = process(nums,1,nums.length - 1,ans);
         //第一个房间选择时，最后一个房间不可以选，
        //所以结束位置nums.length - 2 ，
        int p2 = nums[0] + process(nums,2,nums.length - 2,ans2);
        //选出最大值
        return Math.max(p1,p2);
    }

		/**
		* 递归加缓存
		* index 小偷来到的房间下标
		* end 结束位置
		* ans 缓存
		*/
      public int process(int[]nums,int index,int end,int[]ans){
        //base case 越界时，返回0
        if(index > end){
            return 0;
        }
        //缓存中有数据直接从缓存中拿。
        if(ans[index] != -1){
            return ans[index];
        }
        //选择不偷当前房间时，去下一间继续做选择
        int p1 = process(nums,index + 1,end,ans);
        //选择当前房间时，下一间就不能选了，要去下下一间去选择，所以 index + 2
        int p2 = process(nums,index + 2,end,ans);
        //因为偷当前房间，所以当前房间的值 加上后序结果，
        p2 += nums[index];
        //取最大值
        int aos =  Math.max(p1,p2);
        //结果放进缓存中
        ans[index] = aos;
        return aos;

    }