状态机DP例题

1.介绍

状态机模型是动态规划（dp）的经典模型之一，一般的状态机模型都是拥有有限个状态，这些状态可以相互转换。

下图是一个简单的状态机模型：

2.例题

2.1 例题1

Leetcode.198打家劫舍

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

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

示例 1：

输入：[1,2,3,1]
输出：4
解释：偷窃 1 号房屋 (金额 = 1) ，然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。

示例 2：

输入：[2,7,9,3,1]
输出：12
解释：偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9)，接着偷窃 5
号房屋 (金额 = 1)。偷窃到的最高金额 = 2 + 9 + 1 = 12 。

提示：

1 <= nums.length <= 100
0 <= nums[i] <= 400

问题分析：

代码：

class Solution {
public:
    int rob(vector<int>& price) {
        int n = price.size();
        int f[n+1][2];
        memset(f,0xcf,sizeof f);
        f[0][0] = 0;
        for(int i = 1;i <= n;i++){
            f[i][0] = max(f[i-1][0],f[i-1][1]);
            f[i][1] = f[i-1][0] + price[i-1];
        }
        return max(f[n][0],f[n][1]);
    }
};

2.2 例题2

Leetcode.188）买卖股票的最佳时机 IV
Leetcode.123）买卖股票的最佳时机 III

给定一个整数数组 prices ，它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1：

输入：k = 2, prices = [2,4,1]
输出：2
解释：在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天
(股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4 - 2 = 2 。

示例 2：

输入：k = 2, prices = [3,2,6,5,0,3]
输出：7
解释：在第 2 天 (股票价格 = 2) 的时候买入，在第 3
天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6 - 2 = 4 。
随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3 - 0 = 3 。

提示：

0 <= k <= 100
0 <= prices.length <= 1000
0 <= prices[i] <= 1000

问题分析：

代码：

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int n = prices.size();
        int f[n+1][k+1][2];
        memset(f,0xcf,sizeof f);
        for(int i = 0;i <= n;i++) f[i][0][0] = 0;
        for(int i = 1;i <= n;i++){
            for(int j = 1;j <= k;j++){
                f[i][j][0] = max(f[i-1][j][0],f[i-1][j][1]+prices[i-1]);
                f[i][j][1] = max(f[i-1][j][1],f[i-1][j-1][0] - prices[i-1]);
            }
        }

        int ans = 0;
        for(int j = 0;j <= k;j++) ans = max(ans,f[n][j][0]);
        return ans;
    }
};

买股票3是简化版，也是一样的思路，这里直接给出代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();

        int f[n+1][3][2];
        memset(f,0xcf,sizeof f);
        for(int i = 0;i <= n;i++) f[i][0][0] = 0;
        for(int i = 1;i <= n;i++){
            for(int j = 1;j <= 2;j++){
                f[i][j][0] = max(f[i-1][j][0],f[i-1][j][1] + prices[i-1]);
                f[i][j][1] = max(f[i-1][j][1],f[i-1][j-1][0] - prices[i-1]);
            }
        }

        int ans = 0;
        for(int j = 0;j <= 2;j++) ans = max(ans,f[n][j][0]);
        return ans;
    }
};

2.3 例题3

Leetcode.309最佳买卖股票时机含冷冻期

给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1:

输入: prices = [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

示例 2:

输入: prices = [1]
输出: 0

提示：

1 <= prices.length <= 5000
0 <= prices[i] <= 1000

问题分析：

代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        int f[n+1][3];
        memset(f,0xcf,sizeof f);
        f[0][2] = 0;
        for(int i = 1;i <= n;i++){
            f[i][0] = max(f[i-1][0],f[i-1][2] - prices[i-1]);
            f[i][1] = f[i-1][0]  prices[i-1];
            f[i][2] = max(f[i-1][1],f[i-1][2]);
        }

        return max(f[n][1],f[n][2]);
    }
};