Leecode #122 买卖股票的最佳时机 II
难度:easy
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例:
示例 1:
输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。
解题思路
如果我可以预测股市,第一件事情是就是告诉我妈千万别入原油的股…anyway,这道题的关键是买入卖出的时间点,以及赚到最多的钱。一开始肯定会想复杂的,所以人要贪心,看着眼前就好了。
什么时候买入?如果明天比今天高,那就买入,因为这样一定能赚(不管赚多赚少)
什么时候卖出?如果明天的价格比今天低,那就赶紧卖掉(今天的价格肯定比买入时的价格高),如果明天的价格比今天高,那就等到明天再看吧!
代码如下:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
l = len(prices)
i = 0
hold = -1
earn = 0
while i < l-1:
if hold == -1: # 没有买入
if prices[i] < prices[i+1]: # 有利可图
hold = prices[i]
else: # 买入了
if prices[i] >prices[i+1]: # 值得卖掉
earn += prices[i] - hold
hold = -1
i += 1
if hold != -1 and hold < prices[-1]:
earn += prices[-1] - hol
return earn