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

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 两次交易所能获得的最大收益
     * @param prices int整型一维数组 股票每一天的价格
     * @return int整型
     */
    public int maxProfit (int[] prices) {
        if(prices == null || prices.length <= 1) return 0 ;
        int n = prices.length ;
        int f[][] = new int[n][5] ;
        f[0][0] = 0 ;//f[i][0]:到第i天为止，没买过股票的最大收益
        f[0][1] = -prices[0] ;//f[i][1]:到第i天为止，买过一次的最大收益
        f[0][2] = f[0][0] ;//f[i][2]:到第i天为止，买过一次，买过一次的最大收益
        f[0][3] = f[0][1] ;//f[i][3]:到第i天为止，买过两次，买过一次的最大收益
        f[0][4] = f[0][0] ;//f[i][4]:到第i天为止，买过两次，买过两次的最大收益
        for(int i = 1 ; i < n ; i ++) {
            f[i][0] = f[i-1][0] ;
            f[i][1] = Math.max(f[i-1][1] , f[i-1][0] - prices[i]) ;
            f[i][2] = Math.max(f[i-1][2] , f[i-1][1] + prices[i]) ;
            f[i][3] = Math.max(f[i-1][3] , f[i-1][2] - prices[i]) ;
            f[i][4] = Math.max(f[i-1][4] , f[i-1][3] + prices[i]) ;
        }
        int max = Math.max(f[n-1][2] , f[n-1][4]) ;
        return  max > 0 ? max : 0 ;
    }
}