leetcode 121 买卖股票的最佳时机

题目

给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。

你只能选择 某一天 买入这只股票，并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。

返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0 。

示例

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

解析

这道题用暴力、贪心、动态规划都能解，在这里就只说动态规划的方法了，还是先用动规五部曲走一遍:
注意本题只可以买入卖出一次
1.确定dp数组及其含义
这里需要定义一个二维DP数组“
dp[i][0] 表示第i天持有股票所得最多现金
dp[i][1] 表示第i天不持有股票所得最多现金
2.确定dp公式
dp[i][0] = max(dp[i - 1][0], -prices[i]);
dp[i][1] = max(dp[i - 1][1], prices[i] + dp[i - 1][0]);
3.dp数组初始化
dp[0][0] = -prices[0];
dp[0][1] = 0;

func maxProfit(prices []int) int {
    // 动态规划解法
    dp := make([][]int, len(prices))
    for i := 0; i < len(prices); i++ {
        dp[i] = make([]int, 2)
    }
    // 所以在这里初始化的是一个二维数组，且第二维度只有0和1两个值
    // 0代表持有该股票，1代表不持有该股票

    // 初始化DP数组
    dp[0][0] = -prices[0] // 若第一天就持有的话，只能是当天买入
    dp[0][1] = 0 // 若第一天不持有，很合理
    for i := 1; i < len(prices); i++ {
        dp[i][0] = max(dp[i-1][0], -prices[i])
        dp[i][1] = max(dp[i-1][1], dp[i-1][0] + prices[i])
    }
    return dp[len(prices)-1][1]
}

func max(a, b int) int {
    if a > b {
        return a
    } 
    return b
}