LeetCode Best Time to Buy and Sell Stock II

LeetCode解题之Best Time to Buy and Sell Stock II

原题

给定每天的股票价格，如果允许进行多次交易，即可以多次买入和卖出，但手中最多只能持有一支股票，在再次买入的时候必须将之前的股票卖出，求能获取的最大利润。

注意点：

无

例子:

输入: prices = [2, 4, 6, 1, 3, 8, 3]

输出: 11([2,6]、[1,8]是两次进行买入卖出的时机)

解题思路

可以进行多次交易的话，为了获取最多的利润，应该在每一段价格上升的区间的开头买入，末尾卖出。从前往后遍历数组，如果价格下跌，则在前一天卖出，在下跌的那天再次买入。不要忘记最后的上升段之后没有下跌的情况，要额外加上。

AC源码

class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
if not prices:
return 0
low = high = prices[0]
profit = 0
for i in range(1, len(prices)):
if prices[i] >= prices[i - 1]:
high = prices[i]
else:
profit += high - low
low = high = prices[i]
profit += high - low
return profit

if __name__ == "__main__":
assert Solution().maxProfit([2, 4, 6, 1, 3, 8, 3]) == 11

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。