动态规划解决股票问题总结

123题和122题对比

122题：在每一天，你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买，然后在 同一天 出售。
123题：给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

题目分析

122题说的是每一天，实际上就代表着你最多可以完成n笔交易，而123题是最多2笔交易。

关键代码对比

122题

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]);//今天不持有股票

123题

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]);//第二次不持有

122题代码实际是化简版，如果我们按照123题的思路来走
代码如下：

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]);//第二次不持有
.
.
.
dp[i][k-1] = Math.max(dp[i-1][k-1],dp[i-1][k-2]+prices[i]);//第k次持有
dp[i][k] = Math.max(dp[i-1][k],dp[i-1][k-1]+prices[i]);//第k次不持有

上面实际上就是通式，这里为什么第122题会简写，其实我觉得我们大可以大胆一点，大家都知道数学里面有一种方法叫做缩放吧，其实我们可以把至多n笔看成是每天都至多1笔，只要我保证了每天持有股票和不持有股票，那我也可以保证每天至多n笔。通过121和122我们可以知道一个通式在1<k<n的情况下可以采取通式来解决股票问题。