题解 | #买卖股票的最好时机(三)#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 两次交易所能获得的最大收益
     * @param prices int整型一维数组 股票每一天的价格
     * @return int整型
     */
    public int maxProfit (int[] prices) {
        // write code here
        // 状态定义dp[i][5]：表示每天可能存在的5种买卖状态
        // dp[i][0]表示到第i天为止没有买过股票的最大收益
        // dp[i][1]表示到第i天为止买过一次股票还没有卖出的最大收益
        // dp[i][2]表示到第i天为止买过一次也卖出过一次股票的最大收益
        // dp[i][3]表示到第i天为止买过两次只卖出过一次股票的最大收益
        // [i][4]表示到第i天为止买过两次同时也买出过两次股票的最大收益
        int dp[][] = new int[prices.length][5];

        // 初始化
        Arrays.fill(dp[0],-100000);

        dp[0][0] = 0;
        dp[0][1] = -prices[0];

        // 状态转移
        for(int i = 1; i < prices.length;i++){
            dp[i][0] = dp[i - 1][0];
            dp[i][1] = Math.max(dp[i - 1][1],dp[i - 1][0] - prices[i]);
            dp[i][2] = Math.max(dp[i - 1][2],dp[i - 1][1] + prices[i]);
            dp[i][3] = Math.max(dp[i - 1][3],dp[i - 1][2] - prices[i]);
            dp[i][4] = Math.max(dp[i - 1][4],dp[i - 1][3] + prices[i]);
        }

        return Math.max(dp[prices.length - 1][2],Math.max(dp[prices.length - 1][4],0));
    }
}