DP表

题目

309. 最佳买卖股票时机含冷冻期
     给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​

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

你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
示例:

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

思路

DP表问题首先分清“状态”和“选择”，然后画出“状态转移框架”，最后写出“状态转移方程”

状态：

0：有股票，非冷冻状态
1：无股票，冷冻状态
2：无股票，非冷冻状态

选择：

买，卖，保持

状态转移框架：

以红线为例，该条状态转移的意义为：第i-1天结束时的状态为 “2.无股票，非冷冻“， 第i天结束时，由于 “买” 股票，状态变为 “0.有股票，非冷冻“

状态转移方程

状态0：
DP(i,0) = max( DP(i-1,0) , DP(i-1,2)-prices(i) )

状态1：
DP(i,1) = DP(i-1,0) + prices(i)

状态2：
DP(2) = max( DP(i-1,2) , DP(i-1,1) )

class Solution:
    def maxProfit(self, prices) -> int:
        f=[float("-inf"),float("-inf"),0]
        for i in range(len(prices)):
            f2=[0,0,0]
            f2[0]=max(f[0],f[2]-prices[i])
            f2[1]=f[0]+prices[i]
            f2[2]=max(f[2],f[1])
            f=f2
        return max(f)

随笔

贪心算法：从前向后的算法
动态规划：从后向前的算法，今天的最大收益 = 前若干天的最大收益 + 前若干天向今天转移的代价