初级算法-买卖股票的最佳时机-Python2

# -*- 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