买卖股票的最佳时机

买卖股票的最佳时机 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。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

大神labuladong已经给出了很详细的解答。
设置一个三维dp状态数组dp[i][k][l]，表示第i天已经进行了k次买入操作（即交易次数），l取值为0,1表示现在是否持有股票。状态转移方程如下所示：
dp[i][k][0] = max(dp[i-1][k][0], dp[i-1][k][1] + prices[i])
表示第i天已经进行了k次买入操作现在没有持有股票可以获得的最大利润。它有两种情况：1.昨天就没有股票，今天和昨天状态一样；2.昨天持有股票，但今天以price[i]的价格把它卖了。

dp[i][k][1] = max(dp[i-1][k][1], dp[i-1][k-1][0] - prices[i])
表示第i天已经进行了k次买入操作现在持有股票可以获得的最大利润。它有两种情况：1.昨天也持有股票，今天和昨天状态一样；2.昨天没有持有股票，但今天以price[i]的价格买入了。

边界情况：
当i=0时（第0天），如果此时没有持有股票，则利润为0，即dp[0][k][0]=0；如果此时持有股票，则利润为负值，这里把它设为-price[0]，即dp[0][k][1]=-price[0]。

最终返回第n-1天进行了k次操作后此时没有持有股票的利润即为最大利润。

代码（c++）

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

买卖股票的最佳时机 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 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

思路与上一题一样。不过测试样例会有一个k很大的情况，dp数组会超内存。一次交易由买入和卖出构成，至少需要两天。所以说有效的k 应该不超过 n/2，如果超过，就相当于 k 没有限制。

代码（c++）

class Solution {
public:
    int maxProfit(vector<int>& prices){
        int dp=0;
        for(int i=1;i<prices.size();i++){
            dp=max(dp,dp+(prices[i]-prices[i-1]));
        }
        return dp;
    }
    int maxProfit(int k, vector<int>& prices) {
        if(prices.size()==0||prices.size()==1) return 0;
        if(k>prices.size()/2) return maxProfit(prices);
        int dp[prices.size()][k+1][2];
        memset(dp,0,sizeof(dp));
        for(int i=0;i<prices.size();i++){
            //for(int j=min(i+1,k);j>=1;j--){
            for(int j=1;j<=k;j++){
                if(i==0){
                    dp[i][j][1]=-prices[i];
                    continue;
                }
                dp[i][j][0]=max(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
                dp[i][j][1]=max(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
                //printf("%d %d\n",dp[i][j][0],dp[i][j][1]);
            }
        }
        return dp[prices.size()-1][k][0];
    }
};