【LeeCode】309. 最佳买卖股票时机含冷冻期

时间:2023-02-16 07:17:44

【题目描述】

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

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

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

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

​https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown/​


【示例】

【LeeCode】309. 最佳买卖股票时机含冷冻期


【代码】​​官方​

package com.company;

// 2022-02-15
class Solution {
public int maxProfit(int[] prices) {
if (prices.length == 0) {
return 0;
}

int n = prices.length;
// f[i][0]: 手上持有股的最大收益
// f[i][1]: 手上不持有股,并且处于冷冻期中的累计最大收益
// f[i][2]: 手上不持有股,并且不在冷冻期中的累计最大收益
int[][] f = new int[n][3];
f[0][0] = -prices[0];
for (int i = 1; i < n; ++i) {
f[i][0] = Math.max(f[i - 1][0], f[i - 1][2] - prices[i]);
f[i][1] = f[i - 1][0] + prices[i];
f[i][2] = Math.max(f[i - 1][1], f[i - 1][2]);
}
return Math.max(f[n - 1][1], f[n - 1][2]);
}
}

public class Test {
public static void main(String[] args) {
new Solution().maxProfit(new int[] {1,2,3,0,2}); // 输出:3
new Solution().maxProfit(new int[] {1}); // 输出: 0
}
}