Leetcode 121. 买卖股票的最佳时机
1.题目描述
给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0 。
示例 1:
输入:[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
示例 2:
输入:prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 没有交易完成, 所以最大利润为 0。
提示:
- 1 <= prices.length <= 10^5
- 0 <= prices[i] <= 10^4
2.我的想法
1.首先按着常规想法,还是想的是暴力输出。
1)先得出数组长度,然后遍历数组。
2)两层循环,第一层就是正常的遍历数组,第二层就是遍历第一层循环的当前元素后的值。
3)之后判断当前元素后的值是否比当前元素的值大,并且两者的差要比之前求的的最大利润大。
4)结束两层循环后输出最后得到的最大利润。
class Solution:
def maxProfit(self, prices: List[int]) -> int:
listlen = len(prices)
maxPrice = 0
for i in range(listlen-1):
for j in range(i+1,listlen):
if prices[j] > prices[i] and prices[j]-prices[i] > maxPrice:
maxPrice = prices[j]-prices[i]
else:
continue
return maxPrice
于是超时了,通过了199 / 212 个通过的测试用例。
2.看了题解,一次遍历即可
1)设置最小的价格为无穷大,最大利润为0(即设置初始值)。
2)遍历数组。如果第i天卖出股票,则最大利润为:该天的股价-前面天数中最小的股价。与已知的最大利润比较,如果大于则更新当前最大利润的值。
3)最后返回遍历数组后得出的最大利润的值。
class Solution:
def maxProfit(self, prices: List[int]) -> int:
minprice = float("inf")
profit = 0
for price in prices:
minprice = min(price, minprice)
profit = max(profit, price - minprice)
return profit
参考Krahets的121. 买卖股票的最佳时机(动态规划,清晰图解)