买卖股票的最佳时机4----最多可交易k次

//k 为任意值的情况
    public static int maxProfit02(int k, int [] prices) {
        if(k <= 0) {
            return 0;
        }

        if(prices.length < 2) {
            return 0;
        }

        if(k > prices.length / 2){
            return maxProfitOfInfinity(prices);
        }

        // 第一维k个状态:k次交易
        // 第二维2个状态:0(卖出股票)、1(买入股票)
        int [][] dp = new int[k][2];
        for(int j = 0; j < k; j ++) {
            dp[j][1] = Integer.MIN_VALUE;
        }

        for(int price : prices) {
            for(int j = 0; j < k; j++){
                if(j == 0) {
                    dp[j][1] = Math.max(dp[j][1], -price);
                } else {
                    //今天我持有着股票,要么我昨天就持有着股票,然后今天选择 rest我昨天本没有持有,但今天我选择 buy
                    dp[j][1] = Math.max(dp[j][1], dp[j - 1][0] - price);
                }
                //今天我没有持有股票
                //昨天就没有持有,然后今天选择 rest,我昨天持有股票,但是今天我 sell
                dp[j][0] = Math.max(dp[j][0], dp[j][1] + price);

            }
        }

        return dp[k - 1][0];
    }


    //对k没有限制,如果k > prices.length / 2 , k 的限制相当于无
    public static int maxProfitOfInfinity(int [] prices) {
        int n = prices.length;
        if(n < 2) {
            return 0;
        }

        int dp_i0 = 0;
        int dp_i1 = Integer.MIN_VALUE;

        for(int price : prices) {
            //今天我没有持有股票
            //昨天就没有持有,然后今天选择 rest,我昨天持有股票,但是今天我 sell
            dp_i0 = Math.max(dp_i0, dp_i1 + price);
            //今天我持有着股票,要么我昨天就持有着股票,然后今天选择 rest我昨天本没有持有,但今天我选择 buy
            dp_i1 = Math.max(dp_i1, dp_i0 - price);
        }

        return dp_i0;
    }
©️2020 CSDN 皮肤主题: 游动-白 设计师:上身试试 返回首页