给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。
注意你不能在买入股票前卖出股票。
方法一:暴力法
两个for循环,但LeetCode上超时。
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
n = len(prices)
profit = 0
for index, price in enumerate(prices):
print(index,'--', price)
tmp_price = price
for i in range(n-1, index, -1):
if prices[i] > tmp_price:
tmp_price = prices[i]
if profit < (tmp_price - price):
profit = tmp_price - price
return profit
a=Solution()
print(a.maxProfit([7,6,4,3,1]))
方法二:
import sys
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
n = len(prices)
max_profit = 0
min_price = sys.maxsize
# 由于只能进行一次交易,那么只要找到所以价格中的最大价格和最小价格即可
for i in range(0, n):
if prices[i] < min_price:
min_price = prices[i]
if prices[i] - min_price > max_profit:
max_profit = prices[i] - min_price
return max_profit
a=Solution()
print(a.maxProfit([7,1,5,3,6,4]))