lincode 打劫房屋

描述

假设你是一个专业的窃贼，准备沿着一条街打劫房屋。每个房子都存放着特定金额的钱。你面临的唯一约束条件是：相邻的房子装着相互联系的防盗系统，且 当相邻的两个房子同一天被打劫时，该系统会自动报警。

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

样例

样例 1:

输入: [3, 8, 4]
输出: 8
解释: 仅仅打劫第二个房子.
样例 2:

输入: [5, 2, 1, 3]
输出: 8
解释: 抢第一个和最后一个房子

挑战

O(n) 时间复杂度 且 O(1) 存储。

思路

设res[i]表示从i到n-1这一段能拿到的最多的钱，那么res[i]一共有两种情况：

1. 拿A[i]: 那么res[i] = A[i] + res[i+2]
2. 不拿A[i]: 那么res[i] = res[i+1]

所以res[i] = max(A[i] + a[i+2], a[i+1])
需要初始化的是res[n-1] = A[n-1] 以及res[n-2] = max(A[n-1], A[n-2])
最终的结果是res[0]

代码

class Solution {
public:
    /**
     * @param A: An array of non-negative integers
     * @return: The maximum amount of money you can rob tonight
     */
    long long houseRobber(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]);
        vector<long long> res(n, 0);
        res[n-1] = A[n-1];
        res[n-2] = max(A[n-1], A[n-2]);
        for (int i = n-3; i >= 0; i--)
            res[i] = max(A[i] + res[i+2], res[i+1]);
        return res[0];
    }
};