lintcode 392. 打劫房屋动态规划

Love The Way You Lie 2023-07-07 15:58 3阅读 0赞

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

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

样例
    样例 1:
    
    输入: [3, 8, 4]
    输出: 8
    解释: 仅仅打劫第二个房子.
    样例 2:
    
    输入: [5, 2, 1, 3] 
    输出: 8
    解释: 抢第一个和最后一个房子
    挑战
    O(n) 时间复杂度 且 O(1) 存储。

思路：利用动态规划，dp【0】=0 ，dp【1】=A【0】，dp【2】=【1】  
目前自己打劫的这栋房屋后，能得到的最大金额必然是打劫前两栋后的金额加上现房屋的金额，即dp【i】=dp【i-2】+A【i-1】  
如果不打劫的话，得到的最大金额就是打劫前一栋房屋后的金额，即dp【i】=dp【i-2】，至于要不要打劫，则判断两种情况哪种得到的金额大即可

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 houseRobber
            long long res=0;
            vector<long long >dp(A.size()+1,0);
            dp[1]=A[0];
            for (int i = 2; i <= A.size(); i++) {
        
                dp[i]=max(dp[i-2]+A[i-1],dp[i-1]);
            }
            return dp[A.size()];
        }
    };