买卖股票的最佳时机iv

73 阅读1分钟
// 买卖股票的最佳时机iv  
// 输入:k = 2, prices = [2,4,1]  
// 输出:2  
// 解释:在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。  
public static int maxProfit(int k,int[]prices){  
    if(prices.length==0)return 0;  
    int n=prices.length;  
    k=Math.min(k,n/2);  
    int[][] buy=new int[n][k+1];  
    int[][] sell=new int[n][k+1];  
    buy[0][0]=-prices[0];  
    sell[0][0]=0;  
    for (int i = 1; i <=k ; i++) {  
        buy[0][i]=sell[0][i]=Integer.MIN_VALUE/2;  
    }  

    for (int i = 1; i <n ; i++) {  
        buy[i][0]=Math.max(buy[i-1][0],sell[i-1][0]-prices[i]);  
        for (int j = 1; j <=k; j++) {  
            buy[i][j]=Math.max(buy[i-1][j],sell[i-1][j]-prices[i]);  
            sell[i][j]=Math.max(sell[i-1][j],buy[i-1][j-1]+prices[i]);  
        }  
    }  
    return Arrays.stream(sell[n-1]).max().getAsInt();  
}