TOP150-LC122-股票问题2

/*股票问题2
给你一个整数数组 prices ，其中 prices[i] 表示某支股票第 i 天的价格。

        在每一天，你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买，然后在 同一天 出售。

        返回 你能获得的 最大 利润 。*/
/*股票系列问题状态定义：

        dp[i][k][0 or 1]
        0 <= i <= n - 1, 1 <= k <= K
        n 为天数，大 K 为交易数的上限，0 和 1 代表是否持有股票。
        股票系列问题通用状态转移方程和 base case：


        状态转移方程：
        dp[i][k][0] = max(dp[i-1][k][0], dp[i-1][k][1] + prices[i])
        dp[i][k][1] = max(dp[i-1][k][1], dp[i-1][k-1][0] - prices[i])

        base case：
        dp[-1][...][0] = dp[...][0][0] = 0
        dp[-1][...][1] = dp[...][0][1] = -infinity
        特化到 k 无限制的情况，状态转移方程如下：

        dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i])
        dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i])*/

public class lc122 {
    public int maxProfit(int[] prices) {
        int n = prices.length;
        int[][] dp = new int[n][2];
        for (int i = 0; i < n; i++) {
            if (i == 0) {
                dp[i][0] = 0;
                dp[i][1] = -prices[i];
            } else {
                dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
                dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] -prices[i]);
            }
        }
        return dp[n - 1][0];
    }
}