LeetCode 3013.将数组分成最小总代价的子数组 II：两个堆维护k-1小 + 滑动窗口

【LetMeFly】3013.将数组分成最小总代价的子数组 II：两个堆维护k-1小 + 滑动窗口

力扣题目链接：https://leetcode.cn/problems/divide-an-array-into-subarrays-with-minimum-cost-ii/

给你一个下标从 0 开始长度为 n 的整数数组 nums 和两个正整数 k 和 dist 。

一个数组的代价是数组中的 第一个 元素。比方说，[1,2,3] 的代价为 1 ，[3,4,1] 的代价为 3 。

你需要将 nums 分割成 k 个 连续且互不相交 的子数组，满足第二个子数组与第 k 个子数组中第一个元素的下标距离 不超过 dist 。换句话说，如果你将 nums 分割成子数组 nums[0..(i₁ - 1)], nums[i₁..(i₂ - 1)], ..., nums[i_k-1..(n - 1)] ，那么它需要满足 i_k-1 - i₁ <= dist 。

请你返回这些子数组的最小总代价。

示例 1：

复制代码

输入：nums = [1,3,2,6,4,2], k = 3, dist = 3
输出：5
解释：将数组分割成 3 个子数组的最优方案是：[1,3] ，[2,6,4] 和 [2] 。这是一个合法分割，因为 ik-1 - i1 等于 5 - 2 = 3 ，等于 dist 。总代价为 nums[0] + nums[2] + nums[5] ，也就是 1 + 2 + 2 = 5 。
5 是分割成 3 个子数组的最小总代价。

示例 2：

复制代码

输入：nums = [10,1,2,2,2,1], k = 4, dist = 3
输出：15
解释：将数组分割成 4 个子数组的最优方案是：[10] ，[1] ，[2] 和 [2,2,1] 。这是一个合法分割，因为 ik-1 - i1 等于 3 - 1 = 2 ，小于 dist 。总代价为 nums[0] + nums[1] + nums[2] + nums[3] ，也就是 10 + 1 + 2 + 2 = 15 。
分割 [10] ，[1] ，[2,2,2] 和 [1] 不是一个合法分割，因为 ik-1 和 i1 的差为 5 - 1 = 4 ，大于 dist 。
15 是分割成 4 个子数组的最小总代价。

示例 3：

复制代码

输入：nums = [10,8,18,9], k = 3, dist = 1
输出：36
解释：将数组分割成 4 个子数组的最优方案是：[10] ，[8] 和 [18,9] 。这是一个合法分割，因为 ik-1 - i1 等于 2 - 1 = 1 ，等于 dist 。总代价为 nums[0] + nums[1] + nums[2] ，也就是 10 + 8 + 18 = 36 。
分割 [10] ，[8,18] 和 [9] 不是一个合法分割，因为 ik-1 和 i1 的差为 3 - 1 = 2 ，大于 dist 。
36 是分割成 3 个子数组的最小总代价。

提示：

3 <= n <= 10⁵
1 <= nums[i] <= 10⁹
3 <= k <= n
k - 2 <= dist <= n - 2

解题方法：有序集合 + 滑动窗口

n u m s nums nums第一个元素必选，剩下 k − 1 k-1 k−1个元素的起始位置间隔必须 ≤ d i s t \leq dist ≤dist。使用一个大小为 d i s t + 1 dist+1 dist+1的滑动窗口，每次求这个窗口中 k − 1 k-1 k−1小元素的和。

问题变成了滑动窗口向右滑动过程中，窗口中不断新增一个元素、移除一个元素的状态下如何保持计算 k − 1 k-1 k−1小的元素有哪些：

我们可以使用两个有序集合，一个叫 s t a g e stage stage代表(正在舞台上的) k − 1 k-1 k−1小元素，一个叫 c a n d i d a t e candidate candidate代表在窗口中但(暂)未被选中的元素。

窗口右移过程中，假设要新加入窗口的元素是 i n in in，移除窗口的元素是 o u t out out，

对于 o u t out out有：

如果 o u t out out在 c a n d i d a t e candidate candidate候选集合中，那么 o u t out out永无上台之日，直接移出候选

如果 o u t out out在 s t a g e stage stage选中集合中，那么 o u t out out是时候退役了，移出 s t a g e stage stage集合，并更新集合中元素之和

对于 i n in in有：

如果 i n in in比 s t a g e stage stage中最大元素小，说明更优秀的人来了，移出 s t a g e stage stage集合中最大的那个，将 i n in in加入 s t a g e stage stage集合，并更新 s t a g e stage stage集合中元素之和

否则，将 i n in in加入候选

之后进行 s t a g e stage stage集合大小的调整：

如果 s t a g e stage stage集合小于 k − 1 k-1 k−1，说明有人退役暂无人顶上，从 c a n d i d a t e candidate candidate中选最小的那个顶上(移出 c a n d i d a t e candidate candidate并加入 s t a g e stage stage)，并更新 s t a g e stage stage集合中元素之和

如果 s t a g e stage stage集合大于 k + 1 k+1 k+1，说明有更优秀的人来了，要把 s t a g e stage stage中最大的那个移出并加入到 c a n d i d a t e candidate candidate，并更新 s t a g e stage stage集合中元素之和

整个滑动窗口移动的过程中，所有 s t a g e stage stage元素和中最小的那个即为答案。

为了方便计算，我们开局直接把 k k k减一。

时间复杂度 O ( n log ⁡ d i s t ) O(n\log dist) O(nlogdist)
空间复杂度 O ( d i s t ) O(dist) O(dist)

AC代码

C++

cpp 复制代码

/*
 * @LastEditTime: 2026-02-03 22:11:11
 */
typedef long long ll;
class Solution {
public:
    ll minimumCost(vector<int>& nums, int k, int dist) {
        k--;
        multiset<ll> stage(nums.begin() + 1, nums.begin() + dist + 2), candidate;
        ll ans = accumulate(nums.begin(), nums.begin() + dist + 2, 0ll);
        while (stage.size() > k) {
                int retiree = *stage.rbegin();
                stage.erase(prev(stage.end()));
                ans -= retiree;
                candidate.insert(retiree);
        }

        ll nowAns = ans;
        for (int end = dist + 2; end < nums.size(); end++) {
            int in = nums[end], out = nums[end - dist - 1];
            
            // out
            multiset<ll>::iterator it = candidate.find(out);
            if (it != candidate.end()) {
                candidate.erase(it);
            } else {
                stage.erase(stage.find(out));
                nowAns -= out;
            }

            // in
            if (in < *stage.rbegin()) {
                stage.insert(in);
                nowAns += in;
            } else {
                candidate.insert(in);
            }

            // resize
            if (stage.size() == k - 1) {
                int newActor = *candidate.begin();
                candidate.erase(candidate.begin());
                stage.insert(newActor);
                nowAns += newActor;
            } else if (stage.size() == k + 1) {
                int retiree = *stage.rbegin();
                stage.erase(prev(stage.end()));
                nowAns -= retiree;
                candidate.insert(retiree);
            }

            ans = min(ans, nowAns);
        }
        return ans;
    }
};

#if defined(_WIN32) || defined(__APPLE__)
/*
[1,3,2,6,4,2]
3
3

5
*/
int main() {
    string s;
    int a, b;
    while (cin >> s >> a >> b) {
        Solution sol;
        vector<int> v = stringToVector(s);
        cout << sol.minimumCost(v, a, b) << endl;
    }
    return 0;
}
#endif

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

千篇源码题解已开源