198.打家劫舍

简介: 198.打家劫舍

198. 打家劫舍


动态规划

初始

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


dp[i]表示前i间房子可以偷取的财物的最大值。首先剔除一间房屋和两间房屋的情况。一件房屋时就偷取这间。两间房屋时偷取其中金额较大的一间。当房屋数大于2时,设房间数为k,有2种选择:


1.偷第k间房屋,那么就不能偷第k-1间,所以总金额为偷取前k-2间房屋最高金额加上偷第k间房屋金额之和。

2.不偷第k间房屋,则总金额为偷取前k-1间房屋的最高金额。

在这2个选项中选择数额较大的,即为偷窃前k间房屋能偷窃的最高总金额。


状态转移方程:

dp[i]=max(dp[i−2]+nums[i−1],dp[i−1])


边界条件:

dp[1]=nums[0]

dp[2]=max(nums[0],nums[1])


优化(滚动数组)

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


考虑到每间房屋最高总金额只与该房屋前两间的最高总金额有关,所以使用滚动数组,每一时刻只用存放前两间房屋的最高总金额。空间复杂度O(1)。

目录
相关文章
|
3月前
代码随想录 Day40 动态规划08 LeetCodeT198打家劫舍 T213打家劫舍II T337 打家劫舍III
代码随想录 Day40 动态规划08 LeetCodeT198打家劫舍 T213打家劫舍II T337 打家劫舍III
29 0
|
3月前
|
Java
leetcode-198:打家劫舍
leetcode-198:打家劫舍
21 0
leetcode-198:打家劫舍
|
3月前
|
Java
leetcode-337:打家劫舍 III
leetcode-337:打家劫舍 III
24 0
|
3月前
|
Java
leetcode-213:打家劫舍 II
leetcode-213:打家劫舍 II
14 0
|
7月前
Leetcode:打家劫舍系列
Leetcode:打家劫舍系列
|
8月前
打家劫舍篇
打家劫舍篇
47 0
|
8月前
leetcode 198. 打家劫舍
leetcode 198. 打家劫舍
|
10月前
|
Java
打家劫舍问题
打家劫舍问题
leetcode 213 打家劫舍II
leetcode 213 打家劫舍II
67 0
leetcode 213 打家劫舍II
leetcode 337 打家劫舍III
leetcode 337 打家劫舍III
62 0
leetcode 337 打家劫舍III