力扣198. 打家劫舍

2023-12-15 05:00:44

动态规划

  • 思路:
    • 寻找状态转移方程:
      • 假设有 n 个房间;
        • 如果偷第 n 个房间,那么第 n - 1 个房间不偷,之前的 n - 2 个房间偷窃到了 M(n - 2),总共可以偷窃到?M(n - 2) + N(n);
        • 如果不偷第 n 个房间,那么 n - 1 个房间可以偷窃到 M(n - 1);
      • 所以,选择其中最大的情况是利益最大化,即 M(n) = max(M(n-2) + N(n), M(n-1));
    • 边界条件:
      • 一个房间的时候 M(1) = N(1)
      • 两个房间的时候 M(2) = max(N(1), N(2))
    • 综上,完整代码:
class Solution {
public:
    int rob(vector<int>& nums) {
        if (nums.empty()) {
            return 0;
        }

        int size = nums.size();
        if (size == 1) {
            return nums[0];
        }

        std::vector<int> dp = std::vector<int>(size, 0);
        dp[0] = nums[0];
        dp[1] = std::max(nums[0], nums[1]);
        for (int i = 2; i < size; ++i) {
            dp[i] = std::max(dp[i - 2] + nums[i], dp[i - 1]);
        }

        return dp[size - 1];
    }
};

文章来源:https://blog.csdn.net/N_BenBird/article/details/134938372
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。