20.动态规划之简单多状态dp问题3

muyierfly

于 2024-07-18 10:25:20 发布

阅读量247

点赞数 7

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

本文链接：https://blog.csdn.net/m0_47017197/article/details/140179140

版权

算法题专栏收录该内容

51 篇文章 0 订阅

订阅专栏

1.按摩师

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

题目解析

算法讲解

编写代码

class Solution {
    public int massage(int[] nums) {
        // 1. 创建 dp 表
        // 2. 初始化
        // 3. 填表
        // 4. 返回值
        int n = nums.length;
        if (n == 0)
            return 0; // 处理边界条件

        int[] f = new int[n];
        int[] g = new int[n];
        f[0] = nums[0];
        for (int i = 1; i < n; i++) {
            f[i] = g[i - 1] + nums[i];
            g[i] = Math.max(f[i - 1], g[i - 1]);
        }
        return Math.max(g[n - 1], f[n - 1]);
    }
}

2.打家劫舍2

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

题目解析

算法讲解

编写代码

class Solution {
    public int rob(int[] nums) {
        int n = nums.length;
        return Math.max(nums[0] + rob1(nums, 2, n - 2), rob1(nums, 1, n - 1));
    }

    public int rob1(int[] nums, int left, int right) {
        if (left > right)
            return 0;
        // 1. 创建 dp 表
        // 2. 初始化
        // 3. 填表
        // 4. 返回
        int n = nums.length;
        int[] f = new int[n];
        int[] g = new int[n];
        f[left] = nums[left];
        for (int i = left + 1; i <= right; i++) {
            f[i] = g[i - 1] + nums[i];
            g[i] = Math.max(g[i - 1], f[i - 1]);
        }
        return Math.max(f[right], g[right]);
    }
}

前情

198. 打家劫舍 - 力扣（LeetCode）

3.删除并获得点数

740. 删除并获得点数 - 力扣（LeetCode）

题目解析

算法讲解

其实就是在预处理后的数组做一次打家劫舍问题

重点是预处理

编写代码

class Solution {
    public int deleteAndEarn(int[] nums) {
        int n = 10001;
        // 1. 预处理
        int[] arr = new int[n];
        for (int x : nums)
            arr[x] += x;

        // 2. dp
        // 创建 dp 表
        int[] f = new int[n];
        int[] g = new int[n];
        // 初始化
        f[0] = arr[0];
        // 填表
        for (int i = 1; i < n; i++) {
            f[i] = g[i - 1] + arr[i];
            g[i] = Math.max(f[i - 1], g[i - 1]);
        }
        // 返回值
        return Math.max(f[n - 1], g[n - 1]);
    }
}

4.粉刷房子

LCR 091. 粉刷房子 - 力扣（LeetCode）

题目解析

算法讲解

编写代码

class Solution {
    public int minCost(int[][] costs) {
        // 1. 创建 dp 表
        // 2. 初始化
        // 3. 填表
        // 4. 返回结果
        int n = costs.length;
        int[][] dp = new int[n + 1][3];
        for (int i = 1; i <= n; i++) {
            dp[i][0] = Math.min(dp[i - 1][1], dp[i - 1][2]) + costs[i - 1][0];
            dp[i][1] = Math.min(dp[i - 1][0], dp[i - 1][2]) + costs[i - 1][1];
            dp[i][2] = Math.min(dp[i - 1][0], dp[i - 1][1]) + costs[i - 1][2];
        }
        return Math.min(dp[n][0], Math.min(dp[n][1], dp[n][2]));
    }
}

5.买卖股票的最佳时期含冷冻期

309. 买卖股票的最佳时机含冷冻期 - 力扣（LeetCode）

题目解析

算法讲解

起始位置表示前一天的状态

箭头表示的是当天的状态

其实不用考虑“买入状态”，因为有股票卖不出去肯定不是最大值

编写代码

class Solution {
    public int maxProfit(int[] prices) {
        // 1. 创建 dp 表
        // 2. 初始化
        // 3. 填表
        // 4. 返回值
        int n = prices.length;
        int[][] dp = new int[n][3];
        dp[0][0] = -prices[0];
        for (int i = 1; i < n; i++) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] - prices[i]);
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][2]);
            dp[i][2] = dp[i - 1][0] + prices[i];
        }
        return Math.max(dp[n - 1][1], dp[n - 1][2]);
    }
}

6.买卖股票的最佳时期含手续费

714. 买卖股票的最佳时机含手续费 - 力扣（LeetCode）

题目解析

算法讲解

其实结果是g(n-1)呢内的

编写代码

class Solution {
    public int maxProfit(int[] prices, int fee) {
        // 1. 创建 dp 表
        // 2. 初始化
        // 3. 填表
        // 4. 返回值
        int n = prices.length;
        int[] f = new int[n];
        int[] g = new int[n];
        f[0] = -prices[0];
        for (int i = 1; i < n; i++) {
            f[i] = Math.max(f[i - 1], g[i - 1] - prices[i]);
            g[i] = Math.max(g[i - 1], f[i - 1] + prices[i] - fee);
        }
        return g[n - 1];
    }
}

7.买卖股票最佳时期3（重点）

123. 买卖股票的最佳时机 III - 力扣（LeetCode）

前情

. - 力扣（LeetCode）买卖股票最佳时期1

. - 力扣（LeetCode）买卖股票最佳时期2

题目解析

算法讲解

1.状态表示

class Solution {
    public int maxProfit(int[] prices) {
        // 1. 创建 dp 表
        // 2. 初始化
        // 3. 填表
        // 4. 返回值
        int INF = 0x3f3f3f3f;
        int n = prices.length;
        int[][] f = new int[n][3];
        int[][] g = new int[n][3];
        for (int j = 0; j < 3; j++)
            f[0][j] = g[0][j] = -INF;
        f[0][0] = -prices[0];
        g[0][0] = 0;
        for (int i = 1; i < n; i++)
            for (int j = 0; j < 3; j++) {
                f[i][j] = Math.max(f[i - 1][j], g[i - 1][j] - prices[i]);
                g[i][j] = g[i - 1][j];
                if (j - 1 >= 0) // 判断状态是否存在
                    g[i][j] = Math.max(g[i][j], f[i - 1][j - 1] + prices[i]);
            }
        int ret = 0; // 赵楚最后⼀⾏的最⼤值
        for (int j = 0; j < 3; j++)
            ret = Math.max(ret, g[n - 1][j]);
        return ret;
    }
}

编写代码

8.买卖股票最佳时期含冷冻期4

188. 买卖股票的最佳时机 IV - 力扣（LeetCode）

题目解析

算法讲解

细节问题

进行优化

编写代码

class Solution {
    public int maxProfit(int k, int[] prices) {
        // 预处理
        int n = prices.length, INF = 0x3f3f3f3f;
        k = Math.min(k, n / 2);
        int[][] f = new int[n][k + 1];
        int[][] g = new int[n][k + 1];
        // 初始化
        for (int j = 0; j <= k; j++)
            f[0][j] = g[0][j] = -INF;
        f[0][0] = -prices[0];
        g[0][0] = 0;
        // 填表
        for (int i = 1; i < n; i++)
            for (int j = 0; j <= k; j++) {
                f[i][j] = Math.max(f[i - 1][j], g[i - 1][j] - prices[i]);
                g[i][j] = g[i - 1][j];
                if (j >= 1)
                    g[i][j] = Math.max(g[i][j], f[i - 1][j - 1] + prices[i]);
            }
        int ret = 0;
        for (int j = 0; j <= k; j++)
            ret = Math.max(ret, g[n - 1][j]);
        return ret;
    }
}