Leecode:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-ii/
'''
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
[7, 1, 5, 6] 第二天买入,第四天卖出,收益最大(6-1),所以一般人可能会想,怎么判断不是第三天就卖出了呢?
这里就把问题复杂化了,根据题目的意思,当天卖出以后,当天还可以买入,所以其实可以第三天卖出,第三天买入,第四天又卖出((5-1)+
(6-5) === 6 - 1)。所以算法可以直接简化为只要今天比昨天大,就卖出。
'''
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
profit = 0
for index, price in enumerate(prices):
if index < len(prices)-1:
if prices[index] < prices[index+1]:
profit += prices[index+1] - prices[index]
return profit
def test_maxProfit():
s = Solution()
assert s.maxProfit([7,1,5,3,6,4]) == 7
assert s.maxProfit([1,2,3,4,5]) == 4
assert s.maxProfit([7,6,4,3,1]) == 0
网友评论