[lintcode] 151 · 买卖股票的蕞佳时机 III

 

描述

假设你有一个数组，它的第i个元素是一支给定的股票在第i天的价格。设计一个算法来找到最大的利润。你最多可以完成两笔交易。

你不可以同时参与多笔交易(你必须在再次购买前出售掉之前的股票)

样例

样例 1

输入 : [4,4,6,1,1,4,2,5]
输出 : 6

思路： 这道题特别适合练习分析状态，考虑n-2到n-1所有可能的状态转移，注意一点就是可以当天买卖直接看做一次买卖就行了，减少不必要的代码量。

public class Solution {
    /**
     * @param prices: Given an integer array
     * @return: Maximum profit
     */
    public int maxProfit(int[] prices) {
        if(prices.length==0) return 0;
        // write your code here
        int[][] dp = new int[prices.length][4];
        //买一次未卖，卖一次，买第二次，卖二次
        dp[0][0] = -prices[0];
        dp[0][2] = -prices[0];
        for(int i=1;i<prices.length;i++){
            dp[i][0] = Math.max(-prices[i], dp[i-1][0]);
            dp[i][1] = Math.max(dp[i-1][0]+prices[i], dp[i-1][1]);
            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]);
        }
 
        return Math.max(dp[prices.length-1][3],dp[prices.length-1][1]) ;
    }
}