买卖股票的最佳时机II
题目:买卖股票的最佳时机II
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出,
这笔交易所能获得利润 = 5-1 = 4 。随后,在第 4 天(股票价格 = 3)的时候买入,
在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。
解法:
求出所有的局部极小值和极大值点,有图可知,求出相邻的极大值与极小值的差值,然后相加,便是最大利润。而如果我们试图跳过其中一个峰值来获取更多利润,那么我们最终将失去其中一笔交易中获得的利润,从而导致总利润的降低。
例如,在上述情况下,如果我们跳过 peak_i和 valley_j,试图通过考虑差异较大的点以获取更多的利润,获得的净利润总是会小与包含它们而获得的净利润,因为 C总是小于 A+B。
代码:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
i = 0
maxprofit = 0
valley = prices[i]
peak = prices[i]
n = len(prices)
while i < n-1:
while i < n-1 and prices[i] >= prices[i+1]:
i += 1
valley = prices[i]
while i < n-1 and prices[i] <= prices[i+1]:
i += 1
peak = prices[i]
maxprofit += peak - valley
return maxprofit