代码随想录Day15 动态规划--4

股票问题

通用的解法都是动态规划 暴力或者贪心都只适用于个别的场景

对于股票问题最重要的是要确定每一天有多少个状态分清状态代码就很清晰

121. 买卖股票的最佳时机

本题只能买一次股票然后再卖一次

我们首先要确定dp数组的含义 我们每一天都可以有两种状态 第一个是持有股票 第二个是不持有该股票

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size()==0){
            return 0;
        }
        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

该题是股票可以多次买入和卖出 但是买入股票之前 前一张股票必须要卖出

所以我们依旧是两种状态 持有和不持有股票 但是我们持有股票的状态可以是之前持有股票的状态保留下来的也可以是之前不持有股票第i天买入后变成持有股票的状态

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size()==0){
            return 0;
        }
        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],dp[i-1][1]-prices[i]);
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]+prices[i]);
        }
        return dp[prices.size()-1][1];
    }
};

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

本题是至多可以买卖两次 所以这里的状态就有4个状态第一次买入卖出和第二次买入卖出

该题初始化的时候 第一次买入没有疑问 第一次卖出同样也是0 为什么第二次也有买入？（可以理解为第一次买入卖出都在当天进行 然后再买入） 同时也就解释了为什么最后是返回最后第二次卖出的状态

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.size() == 0) return 0;
        vector<vector<int>> dp(prices.size(), vector<int>(5, 0));
        dp[0][1] = -prices[0];
        dp[0][3] = -prices[0];
        for (int i = 1; i < prices.size(); i++) {
            dp[i][1] = max(dp[i - 1][1], 0 - 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[prices.size() - 1][4];
    }
};

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

本题是我们最多可以买卖k次股票 根据上一题的思路 我们就有2*k个状态 并且奇数都是买入股票的状态 所以我们初始化的时候 只需要初始化奇数的数组 然后剩余代码都和之前相似 注意返回值

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(prices.size()==0){
            return 0;
        }
        vector<vector<int>> dp(prices.size(),vector<int>(k*2+1,0));
        for(int j=1;j<2*k;j+=2){
            dp[0][j]=-prices[0];
        }
        for(int i=1;i<prices.size();i++){
            for(int j=0;j<2*k-1;j+=2){
                dp[i][j+1]=max(dp[i-1][j+1],dp[i-1][j]-prices[i]);
                dp[i][j+2]=max(dp[i-1][j+2],dp[i-1][j+1]+prices[i]);
            }
        }
        return dp[prices.size()-1][2*k];
    }
};

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

该题可以多次买卖股票 但是你卖出股票后有一天的冷冻期（不能继续买入股票）所以这道题在原有的基础上除了持有和不持有股票两种状态 还应当加入冷冻期的状态 由于冷冻期就是在卖出股票的后一天 所以我们应当加入当天卖出股票的状态

所以我们持有股票的状态可以由保持卖出股票的状态和冷冻期的状态推导而来 由于当天卖出股票和保持卖出股票其实是同一种状态 我们应该先取他们的最大值 再和前一天持有股票的状态作比较

由于当天卖出股票和保持卖出股票其实是同一种状态 所以最后的返回值要先取他们的最大值

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len=prices.size();
        if(len==0) return 0;
        vector<vector<int>> dp(len,vector<int>(4,0));
        dp[0][0]=-prices[0];
        for(int i=1;i<len;i++){
            dp[i][0]=max(dp[i-1][0],max(dp[i-1][3]-prices[i],dp[i-1][1]-prices[i]));
            dp[i][1]=max(dp[i-1][1],dp[i-1][3]);    //保持卖出股票的状态
            dp[i][2]=dp[i-1][0]+prices[i];  //当天卖出股票 便于确定冷冻期
            dp[i][3]=dp[i-1][2];
        }
        return max(dp[len-1][3],max(dp[len-1][1],dp[len-1][2]));
    }
};

714.买卖股票的最佳时机含手续费 

这一道题可以多次买卖股票 只不过一次买卖股票的过程中有一个手续费 只需要在原有的代码基础上减去手续费就行了（简单）

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int len=prices.size();
        if(len==0){
            return 0;
        }
        vector<vector<int>> dp(len,vector<int>(2,0));
        dp[0][0]=-prices[0];
        dp[0][1]=0;
        for(int i=1;i<len;i++){
            dp[i][0]=max(dp[i-1][0],dp[i-1][1]-prices[i]);
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]+prices[i]-fee);
        }
        return dp[len-1][1];
    }
};