# -*- coding:utf-8 -*-
class Solution(object):
"""
给你一个整数数组 prices ,其中 prices[i] 表示某支股票第 i 天的价格。
在每一天,你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买,然后在 同一天 出售。
返回你能获得的 最大 利润 。
"""
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
# 算法
def main():
solution = Solution()
prices = [7, 1, 5, 3, 6, 4]
assert solution.maxProfit(prices) == 7
prices = [1, 2, 3, 4, 5]
assert solution.maxProfit(prices) == 4
prices = [7, 6, 4, 3, 1]
assert solution.maxProfit(prices) == 0
print('通过')
if __name__ == '__main__':
main()
方法:
def maxProfit(self, prices):
""" 原理是,只要当前元素比后一个元素小,就把他们的差算到利润中
:type prices: List[int]
:rtype: int
"""
profit = 0
# 当前持有的股票下标
cur = 0
for i in range(1, len(prices)):
if prices[cur] < prices[i]:
# 卖出
profit += prices[i] - prices[cur]
cur = i
return profit