"""
剑指 Offer 63. 股票的最大利润
假设把某股票的价格按照时间先后顺序存储在数组中，请问买卖该股票一次可能获得的最大利润是多少？

示例 1:

输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天（股票价格 = 1）的时候买入，在第 5 天（股票价格 = 6）的时候卖出，最大利润 = 6-1 = 5 。
     注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。
示例 2:

输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
"""
"""
dp[i][k][0] = max(dp[i-1][k][0], dp[i-1][k][1]+price[i])
dp[i][k][1] = max(dp[i-1][k][1], dp[i-1][k-1][0]-price[i])


k=1
dp[i][0] = max(dp[i-1][0], dp[i-1][1]+price[i])
dp[i][1] = max(dp[i-1][1], -price[i])
"""

class Solution(object):
    def maxProfit(self, prices):
        """
        :type prices: List[int]
        :rtype: int
        """
        n = len(prices)
        if n < 1: return 0
        dp = [[0 for _ in range(2)] for _ in range(n)]
        for i in range(n):
            if i == 0:
                dp[i][0] = 0
                dp[i][1] = -prices[i]
            else:
                dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i])
                dp[i][1] = max(dp[i-1][1], -prices[i])
        return dp[n-1][0]

    def maxProfit1(self, prices):
        n = len(prices)
        if n < 1: return 0
        dp_i_0 = 0
        dp_i_1 = -prices[0]
        for i in range(n):
            dp_i_0 = max(dp_i_0, dp_i_1+prices[i])
            dp_i_1 = max(dp_i_1, -prices[i])
        return dp_i_0

if __name__ == "__main__":
    prices = [7,1,5,3,6,4]
    res = Solution().maxProfit(prices)
    print(res)