LintCode:打劫房屋 II

在上次打劫完一条街道之后,窃贼又发现了一个新的可以打劫的地方,但这次所有的房子围成了一个圈,这就意味着第一间房子和最后一间房子是挨着的。每个房子都存放着特定金额的钱。你面临的唯一约束条件是:相邻的房子装着相互联系的防盗系统,且 当相邻的两个房子同一天被打劫时,该系统会自动报警

给定一个非负整数列表,表示每个房子中存放的钱, 算一算,如果今晚去打劫,你最多可以得到多少钱 在不触动报警装置的情况下。

 注意事项

这题是House Robber的扩展,只不过是由直线变成了圈

样例

给出nums = [3,6,4], 返回 6, 你不能打劫34所在的房间,因为它们围成一个圈,是相邻的.


思路:

分为两种情况:打劫第一个房屋,则打劫范围是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 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           {
             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           {
             now1 = max(p, pp+A[i]);
             pp = p;
             p = now1;
          }
          return max(now, now1);
        }
    }
};

你可能感兴趣的:(lintcode)