动态规划--股票算法

买卖股票的最佳时机

给定一个数组，它的第 i 个元素是一支给定股票第 i 天的价格。

如果你最多只允许完成一笔交易（即买入和卖出一支股票），设计一个算法来计算你所能获取的最大利润。

注意你不能在买入股票前卖出股票。

例题

输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天（股票价格 = 1）的时候买入，在第 5 天（股票价格 = 6）的时候卖出，最大利润 = 6-1 = 5 。
     注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。

状态

有 买入（buy） 和 卖出（sell） 这两种状态。

转移方程

对于买来说，买之后可以卖出（进入卖状态），也可以不再进行股票交易（保持买状态）。

对于卖来说，卖出股票后不在进行股票交易（还在卖状态）。

只有在手上的钱才算钱，手上的钱购买当天的股票后相当于亏损。也就是说当天买的话意味着损失-prices[i]，当天卖的话意味着增加prices[i]，当天卖出总的收益就是 buy+prices[i] 。

所以我们只要考虑当天买和之前买哪个收益更高，当天卖和之前卖哪个收益更高。

• buy = max(buy, -price[i])  （注意：根据定义 buy 是负数）

• sell = max(sell,  prices[i] + buy)

边界

第一天 buy = -prices[0], sell = 0，最后返回 sell 即可。

def func(a):
    if len(a)<=1:
        return 0
    buy,sell=-a[0],0
    for i in range(len(a)):
        buy=max(buy,-a[i])
        sell=max(sell,buy+a[i])
    return sell

a=[7,1,5,3,6,4]
print(func(a))

买卖股票的最佳时机 II 

给定一个数组，它的第 i 个元素是一支给定股票第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易（多次买卖一支股票）。 

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

例题

输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天（股票价格 = 1）的时候买入，在第 3 天（股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
     随后，在第 4 天（股票价格 = 3）的时候买入，在第 5 天（股票价格 = 6）的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。

状态

有 买入（buy） 和 卖出（sell） 这两种状态。

转移方程

对比上题，这里可以有无限次的买入和卖出，也就是说 买入 状态之前可拥有 卖出 状态，所以买入的转移方程需要变化。

• buy = max(buy, sell - price[i])

• sell = max(sell,   buy + prices[i] )

边界

第一天 buy = -prices[0], sell = 0，最后返回 sell 即可。

 

def func(a):
    if len(a)<=1:
        return 0
    buy,sell=-a[0],0
    for i in range(len(a)):
        buy=max(buy,sell-a[i])
        sell=max(sell,buy+a[i])
    return sell

a=[7,1,5,3,6,4]
print(func(a))

买卖股票的最佳时机 III

给定一个数组，它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

注意: 你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）

输入: [3,3,5,0,0,3,1,4]
输出: 6
解释: 在第 4 天（股票价格 = 0）的时候买入，在第 6 天（股票价格 = 3）的时候卖出，这笔交易所能获得利润 = 3-0 = 3 。
     随后，在第 7 天（股票价格 = 1）的时候买入，在第 8 天 （股票价格 = 4）的时候卖出，这笔交易所能获得利润 = 4-1 = 3 。 

状态

有 第一次买入（fstBuy） 、 第一次卖出（fstSell）、第二次买入（secBuy） 和 第二次卖出（secSell） 这四种状态。

转移方程

这里可以有两次的买入和卖出，也就是说 买入 状态之前可拥有 卖出 状态，所以买入和卖出的转移方程需要变化。

• fstBuy = max(fstBuy ，  -price[i])

• fstSell = max(fstSell，fstBuy + prices[i] )

• secBuy = max(secBuy ，fstSell -price[i]) (受第一次卖出状态的影响)

• secSell = max(secSell ，secBuy + prices[i] )

边界

• 一开始 fstBuy = -prices[0]

• 买入后直接卖出，fstSell = 0

• 买入后再卖出再买入，secBuy - prices[0]

• 买入后再卖出再买入再卖出，secSell = 0

 

def func(a):
    if len(a)<=1:
        return 0
    fbuy,fsell=-a[0],0
    sbuy,ssell=-a[0],0
    for i in range(len(a)):
        fbuy=max(fbuy,-a[i])
        fsell=max(fsell,fbuy+a[i])
        sbuy=max(sbuy,fsell-a[i])
        ssell=max(ssell,sbuy+a[i])
    return ssell

a=[3,3,5,0,0,3,1,4]
# a=[1,2,3,4,5]
print(func(a))

 

 参考代码：

https://mp.weixin.qq.com/s?__biz=MzUyNjQxNjYyMg==&mid=2247485288&idx=1&sn=fd043fc723f38bcaecc90d9945981f8a&chksm=fa0e68e9cd79e1ffd965205bb06b1731539bf2e0bbc5991664f5d1d9721b346ec08c85bb9042&scene=21#wechat_redirect