213. House Robber II


  1. 2016/Oct/19,做了不止一次,本次独立解出,大概耗时13分钟

本题并不难,跟198. House Robber解法基本一样,都是用dynamic programming解,需要做一点小改动:

  • dp两次,第一次不算首元素,第二次不算尾元素,求这两次的最大值

代码如下:

class Solution {
public:
    int rob(vector<int>& nums) {
        if (nums.size() == 0) {
            return 0;
        }
        if (nums.size() == 1) {
            return nums[0];
        }
        if (nums.size() == 2) {
            return max(nums[0], nums[1]);
        }
        vector<int> dp1(nums.size() - 1, 0);
        vector<int> dp2(nums.size() - 1, 0);
        dp1[0] = nums[0];
        dp1[1] = max(nums[0], nums[1]);
        for (int i = 2; i < nums.size() - 1; ++i) {
            dp1[i] = max(dp1[i-1], dp1[i-2] + nums[i]);
        }
        dp2[0] = nums[1];
        dp2[1] = max(nums[1], nums[2]);
        for (int i = 3; i < nums.size(); ++i) {
            dp2[i - 1] = max(dp2[i-2], dp2[i-3] + nums[i]);
        }
        return max(dp1.back(), dp2.back());
    }
};

results matching ""

    No results matching ""