当前位置:   article > 正文

力扣 123. 买卖股票的最佳时机 III

力扣 123. 买卖股票的最佳时机 III

题目来源:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-iii/description/

C++题解:动态规划。至多买卖两次,这意味着可以买卖一次,可以买卖两次,也可以不买卖。

一天一共就有四个状态:

第一次持有股票、第一次不持有股票、第二次持有股票、第二次不持有股票

dp[i][j]中 i表示第i天,j为 [0 - 3] 四个状态,dp[i][j]表示第i天状态j所剩最大现金。

需要注意:dp[i][0],表示的是第i天,买入股票的状态,并不是说一定要第i天买入股票

例如 dp[i][0] ,并不是说 第i天一定买入股票,有可能 第 i-1天 就买入了,那么 dp[i][0] 延续买入股票的这个状态。

dp[i][2],第i天第二次持有股票。dp[0][2]怎么初始化,可以视作之前买进卖出到第0天获益为0,所以dp[0][2]=-prices[0]

  1. class Solution {
  2. public:
  3. int maxProfit(vector<int>& prices) {
  4. int n = prices.size();
  5. if(n <= 1) return 0;
  6. // dp[i][j], j为0-3,0第一次买入,1第一次卖出,2第二次买入,3第二次卖出
  7. vector<vector<int>> dp(n, vector<int>(4, 0));
  8. dp[0][0] = -prices[0];dp[0][2] = -prices[0];
  9. for(int i = 1; i < n; i++) {
  10. dp[i][0] = max(dp[i-1][0], -prices[i]);
  11. dp[i][1] = max(dp[i-1][1], dp[i-1][0] + prices[i]);
  12. dp[i][2] = max(dp[i-1][2], dp[i-1][1] - prices[i]);
  13. dp[i][3] = max(dp[i-1][3], dp[i-1][2] + prices[i]);
  14. }
  15. return dp[n-1][3]; // 虽然可能没有第二次卖出,但是第二次卖出的获益一定大于等于第一次获益
  16. }
  17. };
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/122891
推荐阅读
相关标签
  

闽ICP备14008679号