目录
一,题目描述
英文描述
You are given an array prices where prices[i] is the price of a given stock on the ith day.
Find the maximum profit you can achieve. You may complete as many transactions as you like (i.e., buy one and sell one share of the stock multiple times) with the following restrictions:
After you sell your stock, you cannot buy stock on the next day (i.e., cooldown one day).
Note: You may not engage in multiple transactions simultaneously (i.e., you must sell the stock before you buy again).
中文描述
给定一个整数数组prices,其中第 prices[i] 表示第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例与说明
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二,解题思路
参考题解@发小男的女的【非状态机的DP讲解,全新思路,超通俗易懂,包你一遍看懂】下的高赞评论@zsnb,虽然只贴了代码,但思路非常清晰,这里再详细记录一下
对每一天的买卖股票情况来说可以分为四种状态:
- 不持有股票,非今天卖出
- 不持有股票,今天卖出
- 持有股票,非今天买入
- 持有股票,今天买入
所以可以用一个二维的dp数组来记录,第一维记录每天的股票买卖情况,第二维记录以上具体的四种状态对应的利润。
状态转移如下:
dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1]); // 不持有股票,非今天卖出。昨天或更早的时候卖出
dp[i][1] = Math.max(dp[i - 1][2], dp[i - 1][3]) + prices[i];// 不持有股票,今天卖出。说明昨天持有股票,今天卖出
dp[i][2] = Math.max(dp[i - 1][2], dp[i - 1][3]); // 持有股票,非今天买入。说明是昨天或更早买入的
dp[i][3] = dp[i - 1][0] - prices[i]; // 持有股票,今天买入。之前不能持有股票,且昨天不能卖出
获得完整dp状态数组后,从不持有股票的两种状态中,选出最大值即可
三,AC代码
Java
class Solution {
public int maxProfit(int[] prices) {
int n = prices.length;
int[][] dp = new int[n][4];
dp[0][0] = 0;// 不持有股票,非今天卖出
dp[0][1] = 0;// 不持有股票,今天卖出
dp[0][2] = -1 * prices[0];// 持有股票,非今天买入。这里初始化时需要注意,已经持有股票了,虽然含义上来讲不应该初始化为第一天卖出的价格,但dp[i][2]和dp[i][3]都是同时出现并比较,所以对最终结果无影响
dp[0][3] = -1 * prices[0];// 持有股票,今天买入
for (int i = 1; i < n; i++) {
dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1]); // 不持有股票,非今天卖出。昨天或更早的时候卖出
dp[i][1] = Math.max(dp[i - 1][2], dp[i - 1][3]) + prices[i];// 不持有股票,今天卖出。说明昨天持有股票,今天卖出
dp[i][2] = Math.max(dp[i - 1][2], dp[i - 1][3]); // 持有股票,非今天买入。说明是昨天或更早买入的
dp[i][3] = dp[i - 1][0] - prices[i]; // 持有股票,今天买入。之前不能持有股票,且昨天不能卖出
}
return Math.max(dp[n - 1][0], dp[n - 1][1]);// 不持有股票的两种情况,取最大值
}
}
四,解题过程
第一搏
清楚dp的含义和状态转移过程后,写代码就很简单啦