House Robber II"/>
Leetcode House Robber II
题意: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]);}}
};
更好的思路是,遍历两次,一次包含头,不包含尾,一次不包含头,包含尾。直接比较结果。
更多推荐
Leetcode House Robber II
发布评论