leetcode: 买卖股票的最佳时机 III && leetcode:买卖股票的最佳时机 IV && leetcode:买卖股票时期含冷冻期

买卖股票的最佳时机 III:题目描述

给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

注意: 你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1:

输入: [3,3,5,0,0,3,1,4]
输出: 6
解释: 在第 4 天（股票价格 = 0）的时候买入，在第 6 天（股票价格 = 3）的时候卖出，这笔交易所能获得利润 = 3-0 = 3 。
随后，在第 7 天（股票价格 = 1）的时候买入，在第 8 天 （股票价格 = 4）的时候卖出，这笔交易所能获得利润 = 4-1 = 3 。
示例 2:

输入: [1,2,3,4,5]
输出: 4
解释: 在第 1 天（股票价格 = 1）的时候买入，在第 5 天 （股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接连购买股票，之后再将它们卖出。
因为这样属于同时参与了多笔交易，你必须在再次购买前出售掉之前的股票。
示例 3:

输入: [7,6,4,3,1]
输出: 0
解释: 在这个情况下, 没有交易完成, 所以最大利润为 0。

解题思路

1. 首先定义一个数组v，用来存第i天进行一次售出可获得最大利润
2. 然后从后开始，算后面几天售出股票可获得的最大利润
3. 两个利润相加，求最大值，即我们所求的可进行两次交易可获得的最大利润

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

买卖股票的最佳时机 IV：题目描述

给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。

注意: 你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1:

输入: [2,4,1], k = 2
输出: 2
解释: 在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。

示例 2:

输入: [3,2,6,5,0,3], k = 2
输出: 7
解释: 在第 2 天 (股票价格 = 2) 的时候买入，在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。

解题思路

1. 每次交易需要两天，所以如果最大交易次数大于prices.size() / 2就代表任意次
2. 每天有两种状态，一种是手里面持有股票，另一种是没有持有股票
3. 持有股票分为两种情况：一种是前一天没有股票，今天新买了股票。另一种是前一天持有股票，今天什么也没干。
4. 没有持有股票分为两种情况：一种是前一天持有股票，今天给卖了。另一种是前一天也没有持有股票，今天啥事没干，也没持股票。
5. dp[i][k][0]：代表的是第i天，正在进行第k次交易，现在的状态是没有持股票。
6. dp[i][k][1]：代表的时第i天，正在进行第k次交易，现在的状态时持股票。
7. 由3可推断出：dp[i][k][1] = max(dp[i - 1][k][1],dp[i - 1][k - 1][0] - prices[i])，一种是昨天和今天不变，都是在进行第k次交易，状态都是持股票。另一种是前一天完成了第k-1次交易，即将股票卖了，没持有股票 ，减去今天买股票的钱。取两者的较大值。
8. 由4推断出：dp[i][k][0] = max(dp[i - 1][k][0],dp[i - 1][k][1] + prices[i])。一种是和前一天的状态一样，另一种是前一天的状态时在进行k次交易，并且已经买了股票，然后今天将股票卖了。取两者的较大值。
9. 最后结果当然是没有持有股票比持有股票的利润要高，所以应该返回最后一天，进行k次交易，没有持有股票的值。

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(prices.size() < 2 || k == 0) return 0;
        if(k >= prices.size()/2) return maxInf(prices);
        
        int dp[prices.size()][k][2]; 
        for(int i = 0; i < prices.size(); i++){
            for(int j = 0; j < k; j++){
                if(i == 0){
                    //第0天
                    dp[i][j][0] = 0;
                    dp[i][j][1] = -prices[0];
                    continue;
                }
                dp[i][j][0]=max(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
                if(j == 0)
                    //第一次交易
                    dp[i][j][1]=max(dp[i-1][j][1],-prices[i]);
                else
                    dp[i][j][1]=max(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
            }  
        }
        return dp[prices.size() - 1][k - 1][0];
    }
    int maxInf(vector<int> prices){
        if(prices.size() < 2) return 0;
        int dp_i_0 = 0, dp_i_1 = -prices[0];
        for(int i = 1; i < prices.size(); i++){
            int temp = dp_i_0;
            //没有持股票的状态
            dp_i_0 = max(dp_i_0,dp_i_1 + prices[i]);
            //持有股票的状态
            dp_i_1 = max(dp_i_1,temp - prices[i]);
        }
        
        return dp_i_0;
        
    }
};

买卖股票时期含冷冻期：题目描述

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。

示例:

输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

解题思路

和前面无限次买卖股票的类似
只不过今天持有股票包含的两种情况有一定变化：
第一种是昨天就有股票，今天什么事情都没敢，所以今天持有股票
另一种是前天没有持有股票，今天新买了股票。昨天就是冷冻期。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size() < 2) return 0;
        //手里面有两种状态，一种是持有股票，另一种是没有持有股票
        //初始化这两种状态
        //没有持有股票的状态
        int dp_i_0 = 0;
        int dp_i_1 = -prices[0];
        int pre = 0;
        for(int i = 1; i < prices.size(); i++){
            //保存上一次没有持有股票的状态
            int temp = dp_i_0;
            //更新这一次没有持有股票的状态
            dp_i_0 = max(dp_i_0,dp_i_1 + prices[i]);
            //更新这一次持有股票的状态
            dp_i_1 = max(dp_i_1,pre - prices[i]);
            //保存上上次没有持有股票的状态
            pre = temp;    
        }  
        return dp_i_0;
    }
};