买卖股票的最佳时机Ⅱ

要求:

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

示例:

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

(2)输入: [1,2,3,4,5]
输出: 4
解释: 在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接连购买股票,之后再将它们卖出。
因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。

(3)输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

分析:

通过比较相邻两个值的大小,低价买入,高价卖出,然后计算收益,另外考虑不买入和不卖出的特殊情况。

代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class Solution:
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
status = 0
length = len(prices)
result = 0
buy = 0
sell = 0
#当列表为空或者只有一位时,返回0
if length <= 1:
return 0
else:
#遍历列表
for x in range(length-1):
#当状态为未买入,并且后一位的值大于前一位值时,买入,并且把状态改为买入
if status == 0 and prices[x] < prices[x+1]:
buy = prices[x]
status = 1
#当状态为买入,并且后一位的值小于等于前一位值时,卖出,并且把状态改为未买入,累计result
elif status == 1 and prices[x] >= prices[x+1]:
sell = prices[x]
result = result + sell - buy
status = 0 # 正常情况最后状态一定是未买入,即status=0

#用来排除[5,4,3,2,1]这种一直没有买入的情况
if status == 0 and sell == 0:
result = 0
#用来解决[1,2,3,4,5]这种一直没有卖出的情况
elif status == 1:
result = result + prices[length-1] - buy
return result
参考资料:

1、买卖股票的最佳时机 II-Python-LeetCode https://blog.csdn.net/linfeng886/article/details/80070804


买卖股票的最佳时机Ⅱ
https://xiepeng21.cn/posts/43c05810/
作者
Peng Xie
发布于
2018年7月17日
许可协议