【线性DP】打家劫舍II
作者:互联网
【题目链接】
【题目描述】
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。
给定一个代表每个房屋存放金额的非负整数数组,计算你 在不触动警报装置的情况下 ,今晚能够偷窃到的最高金额。
【输入输出样例】
2 3 2
3
【数据范围】
1 <= nums.length <= 100
0 <= nums[i] <= 400
不同与打家劫舍,由于存在环,因此头尾的数据需要特殊处理,当偷窃了第1个房屋,则第n个房屋就一定不能被选择;当第n个房屋被选择,则第1个房屋也一定不能被选择。
那么可以通过直接放弃偷窃第1个房屋或者第n个房屋来计算所能偷窃到的最高金额,之后取两种决策中的最大值即可。
1 #include <algorithm> 2 const int N = 109; 3 int f[N]; 4 class Solution { 5 public: 6 int rob(vector<int>& nums) { 7 if(nums.size() == 1) 8 return nums[0]; 9 if(nums.size() == 2) 10 return max(nums[0],nums[1]); 11 return max(dp(nums,0,nums.size()-2),dp(nums,1,nums.size() - 1)); 12 } 13 int dp(vector<int>& nums,int i,int j) 14 { 15 f[i] = nums[i]; 16 f[i + 1] = max(nums[i],nums[i+1]); 17 for(int k = i + 2;k <= j;++k) 18 f[k] = max(f[k - 1],f[k - 2] + nums[k]); 19 return f[j]; 20 } 21 };
标签:偷窃,nums,int,II,房屋,return,打家劫舍,DP,size 来源: https://www.cnblogs.com/fengshi99/p/15408676.html