代码随想录算法训练营第五十天|123.买卖股票的最佳时机III、188. 买卖股票的最佳时机 IV

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

思路：这道题区分于前两道题，前两道题，分别是可进行一次交易和可进行无数次交易，这道题是可进行两次交易，但共同点是手中仅能握有一只股票！首先确定dp数组有几种状态，五种状态，不买入，第一次买入，第一次卖出，第二次买入，第二次卖出。所以说其实很简单了！按照这个思路写就行了

第一版代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len=prices.size();
        if(len==0)return 0;
        vector<vector<int>> dp(len,vector<int>(5,0));
        //啥也没买
        dp[0][0]=0;
        //第一天买入
        dp[0][1]=-prices[0];
        //第一天卖出（相等于买入又卖出）
        dp[0][2]=0;
        //第一天第二次买入
        dp[0][3]=-prices[0];
        //第一天第二次买出
        dp[0][4]=0;
        //为什么要初始化呢？因为采用i-1，跟前一个状态有关
        for(int i=1;i<len;i++)
        {
            //dp[i][0]=dp[i-1][0];
            //dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]);
            dp[i][1]=max(dp[i-1][1],-prices[i]);
            dp[i][2]=max(dp[i-1][2],dp[i-1][1]+prices[i]);
            dp[i][3]=max(dp[i-1][3],dp[i-1][2]-prices[i]);
            dp[i][4]=max(dp[i-1][4],dp[i-1][3]+prices[i]);
        }
        return dp[len-1][4];
    }
};

第二版代码：

其实就是把一些重复初始化的代码给删掉了，但是我还更加倾向写第一种，每个思路都十分清晰！

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len=prices.size();
        if(len==0)return 0;
        vector<vector<int>> dp(len,vector<int>(5,0));
        //第一天买入
        dp[0][1]=-prices[0];
        //第一天第二次买入
        dp[0][3]=-prices[0];
        //第一天第二次买出
        //为什么要初始化呢？因为采用i-1，跟前一个状态有关
        for(int i=1;i<len;i++)
        {
            dp[i][1]=max(dp[i-1][1],-prices[i]);
            dp[i][2]=max(dp[i-1][2],dp[i-1][1]+prices[i]);
            dp[i][3]=max(dp[i-1][3],dp[i-1][2]-prices[i]);
            dp[i][4]=max(dp[i-1][4],dp[i-1][3]+prices[i]);
        }
        return dp[len-1][4];
    }
};

188. 买卖股票的最佳时机 IV

思路：其实这道题的思路与上一道本质没啥区别，思路也差不多，就是交易次数从固定具体的次数变成了k次！

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int len=prices.size();
        if(len==0)return 0;
        vector<vector<int>> dp(len,vector<int>(2*k+1,0));
        //奇数代表买入，偶数卖出
        for(int i=1;i<2*k;i+=2)
        {
            dp[0][i]=-prices[0];
        }
        for(int i=1;i<len;i++)
        {
            for(int j=1;j<2*k+1;j++)
            {
                if(j%2==1)
                {
                    dp[i][j]=max(dp[i-1][j],dp[i-1][j-1]-prices[i]);
                }
                else
                {
                    dp[i][j]=max(dp[i-1][j],dp[i-1][j-1]+prices[i]);
                }
            }
        }
        return dp[len-1][2*k];
    }
};

买卖股票问题总结：
对于股票问题，首先选择一个二维dp数组，第一个参数是表示第几天的股票，第二个表示买入的次数，然后进行初始化第一天的数据，因为都是从i-1推导出第i的，最后进行遍历！