Leetcode House Robber II

来源：互联网发布：苹果苹方字体下载 mac 编辑：程序博客网时间：2024/06/15 01:38

题意：House Robber 的衍生题，将直线改成环。

思路：遍历两次，一次包含头，一次不包含头。包含头的需要记录是否最终结果包含头。

class Solution {public:    int rob(vector<int>& nums) {        if(nums.size() == 0) return 0;        if(nums.size() == 1) return nums[0];                vector<bool> incf(nums.size(), 0);        vector<int> robs = nums;        vector<int> robf = nums;        if(nums[0] < nums[1]) incf[1] = true;        else robf[1] = robf[0];        for(int i = 3; i < robs.size(); ++ i) {            robs[i] = max(robs[i] + robs[i - 2], robs[i - 1]);        }        for(int i = 2; i < robf.size(); ++ i) {            if(robf[i] + robf[i - 2] > robf[i - 1]) {                robf[i] += robf[i - 2];                incf[i] = incf[i - 2];            }            else{                robf[i] = robf[i - 1];                incf[i] = incf[i - 1];            }        }        int re = 0;        if(incf[nums.size() - 1]) {            for(int i = 0; i < robf.size(); ++ i)                re = max(re, robf[i]);            return re;                    }        else {            return max(robf[nums.size() - 2], robs[nums.size() - 1]);        }            }};

更好的思路是，遍历两次，一次包含头，不包含尾，一次不包含头，包含尾。直接比较结果。

0 0