LeetCode Best Time to Buy and Sell Stock IV

最新推荐文章于 2025-04-18 22:24:21 发布

原创最新推荐文章于 2025-04-18 22:24:21 发布 · 984 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 3 个专栏收录

122 篇文章

订阅专栏

23 篇文章

订阅专栏

Array

20 篇文章

订阅专栏

本文介绍了一种解决LeetCode上“股票买卖最佳时机IV”问题的高效算法。该问题为给定一系列股价和交易次数限制k的情况下，求最大利润。文章提供了一个Java实现的解决方案，当k大于等于股价长度时，采用特殊情况的处理方式，简化了动态规划过程。

原题链接在这里：https://leetcode.com/problems/best-time-to-buy-and-sell-stock-iv/

本题是Best Time to Buy and Sell Stock III的general情况。

但这里要注意若是k很大已经超过了prices.length的时候，若是按照DP的方法做会浪费时间以及空间。

此时可以直接参照Best Time to Buy and Sell Stock II的做法。

AC Java:

public class Solution {
    public int maxProfit(int k, int[] prices) {
        if(prices == null || prices.length == 0){
            return 0;
        }
        if(k>=prices.length){
            int res = 0;
            for(int i = 1; i < prices.length; i++){
                res += Math.max(0, prices[i]-prices[i-1]);
            }
            return res;
        }
        
        int[] local = new int[k+1];
        int[] global = new int[k+1];
        for(int i = 1; i<prices.length; i++){
            int diff = prices[i] - prices[i-1];
            for(int j = k; j>=1; j--){
                local[j] = Math.max(global[j-1] + Math.max(diff,0), local[j] + diff);
                global[j] = Math.max(global[j], local[j]);
            }
        }
        return global[k];
    }
}