LeetCode - 198. 打家劫舍

198. 打家劫舍

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。

LeetCode - 198. 打家劫舍_第1张图片
解题思路: 动态规划,dp[i]表示第i个房间偷或者不偷能得到的最大金额。状态转移方程:dp[i] = max(dp[i - 1], dp[i - 2] + nums[i]).然后再处理一些corner case就基本解题了。

class Solution {
     
public:
    int rob(vector<int>& nums) {
     
        int n = nums.size();
        if (n == 0) return 0;
        else if (n == 1) return nums[0];
        else if (n == 2) return max(nums[0], nums[1]);
        vector<int> dp(n, 0);
        dp[0] = nums[0];
        dp[1] = max(nums[0], nums[1]);
        for (int i = 2; i < n; ++i) {
     
            dp[i] = max(dp[i - 1], dp[i - 2] + nums[i]);
        }
        return dp.back();
    }
};

你可能感兴趣的:(leetcode,leetcode,动态规划)