python练习–买股票的最佳时机2
一.题目描述:
给定一个数组 prices ,其中 prices[i] 是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例:
输入: prices = [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6-3 = 3 。
输入: prices = [1,2,3,4,5]
输出: 4
解释: 在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接连购买股票,之后再将它们卖出。因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。
输入: prices = [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
提示:
1 <= prices.length <= 3 * 104
0 <= prices[i] <= 104
二.解题思路:
方法一双指针(遍历数组):
当我们没有购入股票的时候,一定是等到最低点再去购买,当我们购入股票后,当发明天要降价,那么今天立即抛售,哪怕后天又涨价,我们可以明天再买入,可以使用双指针遍历数组,来获得最大利润;
**方法二一次遍历:**只要今天价格小于明天价格就在今天买入然后明天卖出。
三.代码
class Solution:
def maxProfit(self, prices: List[int]) -> int:
left = 0
right = 1
buy = False
ret = 0
tmp = 0
while right < len(prices):
if not buy:
if prices[right] > prices[left]:
buy = True
tmp = prices[left]
else:
if prices[right] < prices[left]:
ret += (prices[left] - tmp)
tmp = 0
buy = False
right += 1
left += 1
if buy:
ret += (prices[-1] - tmp)
return ret
这个方法会更简便一点(大佬大佬方法)
class Solution:
def maxProfit(self, prices: List[int]) -> int:
ans = 0
for i in range(1, len(prices)):
if prices[i] > prices[i-1]:
ans += prices[i] - prices[i-1]
return ans