力扣:最佳买卖股票时机
最近在力扣上刷到最佳买卖股票时机系列的题型,主要采用动态规划的思想来解决此类问题。在这里从简单情况,慢慢过渡到相对复杂的情况分析。
题一、买卖股票的最佳时机
题目描述
给定一个数组,它的第 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】 有更多的解法
升级版:力扣 -> 最佳买卖股票时机含冷冻期