买卖股票的最佳时机 II

转自：https://www.cnblogs.com/peterzone/p/9097509.html
题目：

给定一个数组，它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易（多次买卖一支股票）。
注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

示例 1:

输入: [7,1,5,3,6,4]
输出: 7
解释: 在第 2 天（股票价格 = 1）的时候买入，在第 3 天（股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。

示例 2:

输入: [1,2,3,4,5]
输出: 4
解释: 在第 1 天（股票价格 = 1）的时候买入，在第 5 天 （股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接连购买股票，之后再将它们卖出。
因为这样属于同时参与了多笔交易，你必须在再次购买前出售掉之前的股票。

示例 3:

输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

题目分析：

简单题，不必赘述。
只强调笔者的思路要点：
首先需要一个标志记录当前股票的状态（下面代码中的flag，0代表空仓，1代表满仓）
其次需要记录买入价格和卖出价格（下面代码中的buy和sale）
最后卖出后算出所得利润（下面代码中的profit）
注意：防止一天交易两次（代码中的continue语句），最后一天要特殊处理。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

' a test module '

__author__ = 'Zhang Shuai'


class Solution:
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        flag = 0
        buy = 0
        sale = 0
        profit = 0
        for i in range(0, len(prices) - 1):
            if flag == 0 and prices[i] < prices[i + 1]:
                buy = prices[i]
                print('buy',buy)
                flag = 1
                continue
            if flag == 1 and prices[i] > prices[i + 1]:
                sale = prices[i]
                print('sale',sale)

                flag = 0
                profit = profit + (sale - buy)
            print('profit', profit)

        if flag == 1:
            sale = prices[len(prices) - 1]
            profit = profit + (sale - buy)
        print('profit', profit)

        return profit

print(Solution().maxProfit([3,7,2,6,3,8,4]))

最后一天要注意的原因：
因为循环的范围是0，len(prices)-2，意思是说循环到prices[-2]就停止了(因为i与i+1比较)，如果这时候还有一个没有卖出，要么是在i=-2的地方买入，能买入一定代表着-1位置可以卖出，所以卖出。如果在-2之前买入的到此还没卖出一定代表着-1比前面的都大，所以卖出。（均是因为我们设置的判断条件）
例子：
[4,1,2]
[4,1,2,3]