简单多状态dp问题

最新推荐文章于 2024-06-12 10:00:23 发布

夹心宝贝

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量622

点赞数 14

分类专栏：动态规划文章标签：动态规划算法

本文链接：https://blog.csdn.net/2302_80873119/article/details/137412353

版权

1、按摩师

面试题 17.16. 按摩师 - 力扣（LeetCode）

1、状态表示：

dp[i]：从前i个顾客预约中，接收预约的最大总时长。

2、状态转移方程：

首先本题特别的地方在于：不能连续接收两个预约，即不能连续选取数组中两个相邻的位置。

那么对于dp[i]：

1、如果接收第i个顾客，那么此时的时长dp[i] = nums[i] + dp[i-2]。则不考虑接收第i-1个顾客的预约。

2、如果不接收第i个顾客，那么此时的时长，就是可以说从前i-1个顾客中选取的最大时长dp[i] = dp[i-1]。

则 dp[i] = max { dp[i-1],nums[i] + dp[i-2] }

3、初始化：

dp[0] = 0，dp[1] = nums[1](下标注意映射)，并且还需要防止越界，做越界检查。

4、填表顺序：

从少到多。

5、返回dp[n]。

class Solution {
public:
    int massage(vector<int>& nums) {
        int n = nums.size();
        vector<int> dp(n+1);
        dp[0] = 0;
        if(n == 0) return dp[0];
        dp[1] = nums[0];
        if(n == 1) return dp[1];
        for(int i = 2;i<=n;i++)
        {
            dp[i] = max(dp[i-1],dp[i-2] + nums[i-1]);//挑选nums[i-1]工作和不挑选nums[i-1]工作的最大值
        }
        return dp[n];
    }
};

2、打家劫舍Ⅱ

213. 打家劫舍 II - 力扣（LeetCode）

这就是一个经典的多状态dp问题：

首先：如果偷第一家，那么就不能偷最后一家。

反之：如果想要偷最后一家，就必须不偷第一家。

1、状态表示：

dp[i]表示：从前i家中方，能够偷取到的最大价值。

2、状态转移方程：

对于第i家：

如果不偷：那么dp[i] = dp[i-1]。

如果偷：那么 dp[i] = nums[i] + dp[i-2]。

则dp[i] = max{dp[i-1] , dp[i-2] + nums[i] }。

3、初始化：

首先dp[0] = 0，无可厚非。

对于dp[1]就要分两类情况了。

如果不偷最后一家，那么就可以偷第一家，dp[1] = nums[1]。

偷最后一家，那么不能偷第一家，dp[1] = 0。

4、填表顺序：

从第一家到最后一家。

5、返回值

返回两种情况下dp[n]的最大值。

class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        vector<int> dp(n+1);//多开一个
        dp[0] = 0;

        //情况1：第一家偷
        dp[1] = nums[0];
        if(n == 1) return dp[1];//处理边界
        for(int i = 2;i<=n-1;i++)//最后一家不偷，只遍历到倒数第二家
        {
            dp[i] = max(dp[i-1],dp[i-2] + nums[i-1]);
        }
        int res1 = dp[n-1];//最后一家不偷

        //情况2：第一家不偷
        dp[1] = 0;
        for(int i = 2;i<=n;i++)
        {
            dp[i] = max(dp[i-1],dp[i-2] + nums[i-1]);
        }
        int res2 = dp[n];//最后一家偷

        return max(res1,res2);
    }
};