力扣309.最佳买股票时机含冷冻期

力扣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

算法思路

dp动态规划：

• 每一天分成三种状态

  • 0：当天持有股票，可以是当天买入的，也可以是之前买入一直未卖出的
  • 1：当天卖出股票（第二天将处于冷冻期）
  • 2：当天没有股票，并且什么也不做（区分于状态0，1）

• 用f[i]表示第i天结束后的最大累计收益

  • 第i天持有股票(0状态)的累计最大收益：f[i][0]=max(f[i−1][0],f[i−1][2]−prices[i])
  • 第i天卖出股票(1状态)的累计最大收益：f[i][1]=f[i−1][0]+prices[i]
  • 第i天处于2状态的累计最大收益：f[i][2]=max(f[i−1][1],f[i−1][2])
  • 最终的累计最大收益：f[n]=max(f[n−1][0],f[n−1][1],f[n−1][2])

代码实现

int maxProfit(int* prices, int pricesSize){
    int f[pricesSize][3];
    memset(f,0,sizeof(int)*3*pricesSize);
    f[0][0]=-prices[0];
    f[0][1]=f[0][2]=0;
    int i,j,k;
    for(i=1;i<pricesSize;i++)
    {
        //持有股票
        f[i][0]=fmax(f[i-1][2]-prices[i],f[i-1][0]);
        //把股票卖了
        f[i][1]=f[i-1][0]+prices[i];
        //啥都没干,也没股票
        f[i][2]=fmax(f[i-1][2],f[i-1][1]);
    }
    return fmax(f[pricesSize-1][1],f[pricesSize-1][2]);
}