在上次打劫完一条街道之后,窃贼又发现了一个新的可以打劫的地方,但这次所有的房子围成了一个圈,这就意味着第一间房子和最后一间房子是挨着的。每个房子都存放着特定金额的钱。你面临的唯一约束条件是:相邻的房子装着相互联系的防盗系统,且 当相邻的两个房子同一天被打劫时,该系统会自动报警。
给定一个非负整数列表,表示每个房子中存放的钱, 算一算,如果今晚去打劫,你最多可以得到多少钱 在不触动报警装置的情况下。
注意事项
这题是House Robber的扩展,只不过是由直线变成了圈
给出nums = [3,6,4]
, 返回 6
, 你不能打劫3
和4
所在的房间,因为它们围成一个圈,是相邻的.
思路:
分为两种情况:打劫第一个房屋,则打劫范围是0至n-2;
不打劫第一个房屋,则把A[0]强行置0,打劫范围为0-n-1;
在这两种情况中取最大值。
class Solution {
public:
/**
* @param nums: An array of non-negative integers.
* return: The maximum amount of money you can rob tonight
*/
long long houseRobber2(vector<int> A) {
// write your code here
int n = A.size();
if(n == 0)
return 0;
if(n == 1)
return A[0];
if(n == 2)
return max(A[0],A[1]);
else
{
//选第一个房屋
long long p ,pp, now=0, now1=0;
pp = A[0];
p = A[0];//正常情况这里是A[1],此处修改为A[0],代表一定选第一个房屋
for(int i=2; i<n; i++)
{
if(i == n-1)
break;
now = max(p, pp+A[i]);
pp = p;
p = now;
}
//不选第一个房屋
pp = 0;//正常情况这里是A[0],此处修改为0,代表不选第一个房屋
p = A[1];
for(int i=2; i<n; i++)
{
now1 = max(p, pp+A[i]);
pp = p;
p = now1;
}
return max(now, now1);
}
}
};