代码随想录算法训练营第五十一天 | 121. 买卖股票的最佳时机、122.买卖股票的最佳时机II

121. 买卖股票的最佳时机

视频讲解： 动态规划之 LeetCode：121.买卖股票的最佳时机1_哔哩哔哩_bilibili

代码随想录

解题思路（股票系列通解）

1.dp含义：因为第i天的股票有两个状态，买和不买，那么就有两个维度

dp[i][0] 第i天持有这支股票的最大现金(不一定买)   dp[i][1] 第i天不持有这支股票的最大现金(不一定卖)，（手上的钱）

2.递推公式

一直保持着持有这支股票的状态 dp[i-1][0]

买入这支股票               -prices[1]

dp[i][0] = max( dp[i-1][0] , -prices[1] );

不持有这支股票一直保持  dp[i-1][1]

如果在第i卖了这支股票  dp[i-1][0] + prices[i]  那么就是第i-1天持股时卖了，加上现金

dp[i][1] = max( dp[i-1][1] , dp[i-1][0] + prices[i] )  

3.初始化

都是由前一天得出来的

dp[0][0] = -prices[0]   持股现金为股票钱

dp[0][0] = 0;   不持股现金为0

4.遍历顺序

从前往后

class Solution {
public:
    int maxProfit(vector<int>& prices) {
      vector<vector<int>> dp(prices.size(),vector<int>(2,0));
      dp[0][0] = -prices[0];    //持股手头的现金
      dp[0][1] = 0;             //不持股手头的现金
      for(int i=1 ; i<prices.size() ; i++)
      {
          dp[i][0]  = max(dp[i-1][0], -prices[i]);   //前者为之前某一天买了的现金，后面是当天买的现金
          dp[i][1] = max( dp[i-1][1], dp[i-1][0] + prices[i]);  //前者为一直保持不持股，后者为如果持股在这天卖了的现金
      }
      return dp[prices.size()-1][1];     //只需要返回不持股的状态，肯定比持股状态的钱多
    }
};

 

122.买卖股票的最佳时机II

视频讲解： https://www.bilibili.com/video/BV1D24y1Q7Ls

https://programmercarl.com/0122.%E4%B9%B0%E5%8D%96%E8%82%A1%E7%A5%A8%E7%9A%84%E6%9C%80%E4%BD%B3%E6%97%B6%E6%9C%BAII%EF%BC%88%E5%8A%A8%E6%80%81%E8%A7%84%E5%88%92%EF%BC%89.html

解题思路

这题当时是使用贪心算法解决的

而这次我们使用动态规划来做

这题卖出股票的思路不变，但是对于买入股票就有所区别了

dp[i][0] = max( dp[i-1][0], )    前者是i-1持股，后者与买入股票1的区别就是，手头是第一次买，现金为0，而这题则是前一天不持股的现金(前面多次买卖的利润) - prices[1] (只能有一支股票)

class Solution {
public:
    int maxProfit(vector<int>& prices) {
            vector<vector<int>> dp(prices.size(),vector<int>(2,0));
            dp[0][0] = -prices[0];   //持股，现金为第一天的价格
            dp[0][1] = 0;   //不持股，此时现金为0
            for(int i=1 ; i< prices.size() ; i++)
            {
                dp[i%2][0] = max( dp[(i-1)%2][0] , dp[(i-1)%2][1] - prices[i] );
                dp[i%2][1] = max( dp[(i-1)%2][1], dp[(i-1)%2][0] + prices[i] );
            }
            return dp[(prices.size()-1)%2][1];    //不持股现金肯定最高
    }
};

时间复杂度：O（n）

空间复杂度：O（1）

收获

继续加油