最近在力扣上刷到最佳买卖股票时机系列的题型,主要采用动态规划的思想来解决此类问题。在这里从简单情况,慢慢过渡到相对复杂的情况分析。

题一、买卖股票的最佳时机



题目描述
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
如果你最多只允许完成一笔交易(即买入和卖出一支股票一次),设计一个算法来计算你所能获取的最大利润。
注意:你不能在买入股票前卖出股票。
输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6 - 1 = 5 。
注意利润不能是 7 - 1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
来源 -> 力扣:买卖股票的最佳时机


题目分析1
1、 状态定义
 状态 dp[i][j] 表示:在下标为 i 的这一天,用户手上持股状态为 j 所获得的最大利润。
 说明:j 只有 2 个值:0 表示不持股(特指卖出股票以后的不持股状态),1 表示持股。
2、 状态转移
 (1) dp[i][0] : 当前不持股,这种状态可以是,昨天不持股转移过来,即dp[i-1][0] ; 也可以是,昨天持股,今天卖出,即dp[i-1][1] + prices[i] ;
 可得 :dp[i][0] = Math.max ( dp[i-1][0] , dp[i-1][1] + prices[i] ) ;
 (2) dp[i][1] : 当前持股,这种状态可以是,昨天持股转移过来,即dp[i-1][1] ; 也可以是 - prices[i] ;
 可得 :dp[i][1] = Math.max ( dp[i-1][1] , - prices[i] ) ;
这里的 - prices[i] 表示,在下标为 i 的这一天,执行买入操作得到的收益。注意:因为题目只允许一次交易,因此不能加上 dp[i - 1][0]
3、 初始情况
 第 0 天不持股,显然 dp[0][0] = 0 ;
 第 0 天持股,显然 dp[0][1] = -prices[0] ;
4、 输出情况
 从状态转移方程可以看出,每一天的状态都考虑了之前的状态。在只发生一次交易的情况下,持有这支股票一定不能使我们获得最大利润。因此输出是 dp[len - 1][0],不可能是持股的状态 dp[len - 1][1]
5、 实现

class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        if (len < 2) {
            return 0;
        }
        // 0:用户手上不持股所能获得的最大利润,特指卖出股票以后的不持股,非指没有进行过任何交易的不持股
        // 1:用户手上持股所能获得的最大利润
        // 注意:因为题目限制只能交易一次,因此状态只可能从 1 到 0,不可能从 0 到 1
        int[][] dp = new int[len][2];
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        for (int i = 1; i < len; i++) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
            dp[i][1] = Math.max(dp[i - 1][1], - prices[i]);
        }
        return dp[len - 1][0];
    }
}

这里发现由于 dp[i] 仅仅依赖于 dp[i - 1] ,因此,还可以进行空间优化。

class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        if (len < 2) {
            return 0;
        }
        //cash 表示不持股时所能获得的最大利润
        //hold 表示持股时所能获得的最大利润
        int cash = 0;
        int hold = -prices[0];
        for (int i = 1; i < len; i++) {
            cash = Math.max(cash , hold + prices[i]);
            hold = Math.max(hold, - prices[i]);
        }
        return cash;
    }
}

题目分析2
打擂法:其实我们发现:只需要关心之前看到的最低股价,于是遍历的过程中,记录下之前看到的最低股价,做差求得最大收益即可。

class Solution {
    public int maxProfit(int[] prices) {
        int day = prices.length;
        if(day < 2)  return 0;
        //保留当前最大的利润
        int profit = 0;
        // 表示在当前位置之前的最小值,假设修正法(打擂台法)
        int minVal = prices[0];
        for(int i = 1;i < day;i ++){
            profit = Math.max(profit,prices[i] - minVal);
            minVal = Math.min(minVal,prices[i]);
        }
        return profit;
    }
}

题二、买卖股票的最佳时机 II



题目描述
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
         随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。
来源 -> 力扣:买卖股票的最佳时机 II


题目分析1
与上面的分析过程一样,唯一不同的是,此处允许多次交易,这就会导致 dp[i][1] 的状态发生改变
即 dp[i][1] = Math.max ( dp[i-1][1] , dp[i-1][0] - prices[i] ) ;

public class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        if (len < 2) {
            return 0;
        }
        // 0:持有现金
        // 1:持有股票
        // 状态转移:0 → 1 → 0 → 1 → 0 → 1 → 0
        int[][] dp = new int[len][2];
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        for (int i = 1; i < len; i++) {
            // 这两行调换顺序也是可以的
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }
        return dp[len - 1][0];
    }
}

同样,这里 dp[i] 仅仅依赖于 dp[i - 1] ,因此,也可以进行空间优化。

public class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        if (len < 2) {
            return 0;
        }
        // cash:持有现金
        // hold:持有股票
        // 状态转移:cash → hold → cash → hold → cash → hold → cash
        int cash = 0;
        int hold = -prices[0];
        int preCash = cash;
        int preHold = hold;
        for (int i = 1; i < len; i++) {
            cash = Math.max(preCash, preHold + prices[i]);
            hold = Math.max(preHold, preCash - prices[i]);
            preCash = cash;
            preHold = hold;
        }
        return cash;
    }
}

题目分析2
其实这题使用动态规划,相对麻烦,一个很简单的思路就是,只要后一天的相对于前一天有收益,即可进行相关的买入和卖出操作。

class Solution {
    public int maxProfit(int[] prices) {
        int res = 0;
        int len = prices.length;
        //只要后一个值大于前一个值,就有收益
        for (int i = 0; i < len - 1; i++) {
            res += Math.max(prices[i + 1] - prices[i], 0);
        }
        return res;
    }
}

注 : 这两道题不适用动态规划的思想,也很简单,可以拿来作为对动态规划分析的过程简单理解
参考博客:【1,2有更多的解法
升级版:力扣 -> 最佳买卖股票时机含冷冻期