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

123

题目：

给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

题目链接：123. 买卖股票的最佳时机 III - 力扣（LeetCode）

思路：

动态规划，区别于前面题，这次有5个状态，还要知道dp[prices.length-1][4]肯定是最大值，因为他包括dp[prices.length-1][2]，如果第一次卖出已经是最大值了，那么我们可以在当天立刻买入再立刻卖出。所以dp[4][4]已经包含了dp[4][2]的情况。也就是说第二次卖出手里所剩的钱一定是最多的。

详情：代码随想录 (programmercarl.com)

代码：

class Solution {
    public int maxProfit(int[] prices) {
        int[][] dp =new int[prices.length][5];
//0 无买卖 1 买了一次没卖 2 买了一次也卖了 3 买了两次卖了一次 4买了两次都卖了
        dp[0][0]=0;
        dp[0][1]=-prices[0];
        dp[0][2]=0;//相当于当天买当天卖
        dp[0][3]=-prices[0];//当天第一次买卖后又买了
        dp[0][4]=0;//第一天两次买卖
        for(int i=1;i<prices.length;i++)
        {
            dp[i][0]=dp[i-1][0];
            dp[i][1] = Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
            dp[i][2] = Math.max(dp[i-1][2],dp[i-1][1]+prices[i]);
            dp[i][3] = Math.max(dp[i-1][3],dp[i-1][2]-prices[i]);
            dp[i][4] = Math.max(dp[i-1][4],dp[i-1][3]+prices[i]);
        }
        return dp[prices.length-1][4];
    }
}

188

题目：

给你一个整数数组 prices 和一个整数 k ，其中 prices[i] 是某支给定的股票在第 i 天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。也就是说，你最多可以买 k 次，卖 k 次。注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

题目链接：188. 买卖股票的最佳时机 IV - 力扣（LeetCode）

思路：

于lc123思路相同，只需要把次数2抽象为次数k即可

详情：代码随想录 (programmercarl.com)

代码：

class Solution {
    public int maxProfit(int k, int[] prices) {
        int[][] dp =new int[prices.length][(2*k)+1];
        for(int i=0;i<(2*k)+1;i++)
        {
            if(i%2==0)  dp[0][i]=0;
            if(i%2==1)  dp[0][i]=-prices[0];
        }
        for(int i=1;i<prices.length;i++)
        {
            for(int j=0;j<(2*k)+1;j++)
            {
                if(j==0)    dp[i][j]=dp[i-1][j];
                else if(j%2==1)     dp[i][j] = Math.max(dp[i-1][j],dp[i-1][j-1]-prices[i]);
                else if(j%2==0)     dp[i][j] = Math.max(dp[i-1][j],dp[i-1][j-1]+prices[i]);
            }
        }
        return dp[prices.length-1][2*k];
    }
}