2462. 雇佣 K 位工人的總代價

2462. 雇佣 K 位工人的總代價 #

給你一個下標從 0 開始的整數數組 costs ，其中 costs[i] 是雇佣第 i 位工人的代價。

同時給你兩個整數 k 和 candidates 。我們想根據以下規則恰好雇佣 k 位工人：

總共進行 k 輪雇佣，且每一輪恰好雇佣一位工人。
在每一輪雇佣中，從最前面 candidates 和最後面 candidates 人中選出代價最小的一位工人，如果有多位代價相同且最小的工人，選擇下標更小的一位工人。
- 比方說，costs = [3,2,7,7,1,2] 且 candidates = 2 ，第一輪雇佣中，我們選擇第 4 位工人，因為他的代價最小。
- 第二輪雇佣，我們選擇第 1 位工人，因為他們的代價與第 4 位工人一樣都是最小代價，而且下標更小。注意每一輪雇佣後，剩餘工人的下標可能會發生變化。
如果剩餘員工數目不足 candidates 人，那麼下一輪雇佣他們中代價最小的一人，如果有多位代價相同且最小的工人，選擇下標更小的一位工人。
一位工人只能被選擇一次。

返回雇佣恰好 k 位工人的總代價。

示例 1：

輸入：costs = [17,12,10,2,7,2,11,20,8], k = 3, candidates = 4
輸出：11
解釋：我們總共雇佣 3 位工人。總代價一開始為 0 。
- 第一輪雇佣，我們從 [17,12,10,2,7,2,11,20,8] 中選擇。最小代價是 2 ，有兩位工人，我們選擇下標更小的一位工人，即第 3 位工人。總代價是 0 + 2 = 2 。
- 第二輪雇佣，我們從 [17,12,10,7,2,11,20,8] 中選擇。最小代價是 2 ，下標為 4 ，總代價是 2 + 2 = 4 。
- 第三輪雇佣，我們從 [17,12,10,7,11,20,8] 中選擇，最小代價是 7 ，下標為 3 ，總代價是 4 + 7 = 11 。注意下標為 3 的工人同時在最前面和最後面 4 位工人中。
總雇佣代價是 11 。

示例 2：

輸入：costs = [1,2,4,1], k = 3, candidates = 3
輸出：4
解釋：我們總共雇佣 3 位工人。總代價一開始為 0 。
- 第一輪雇佣，我們從 [1,2,4,1] 中選擇。最小代價為 1 ，有兩位工人，我們選擇下標更小的一位工人，即第 0 位工人，總代價是 0 + 1 = 1 。注意，下標為 1 和 2 的工人同時在最前面和最後面 3 位工人中。
- 第二輪雇佣，我們從 [2,4,1] 中選擇。最小代價為 1 ，下標為 2 ，總代價是 1 + 1 = 2 。
- 第三輪雇佣，少於 3 位工人，我們從剩餘工人 [2,4] 中選擇。最小代價是 2 ，下標為 0 。總代價為 2 + 2 = 4 。
總雇佣代價是 4 。

提示：

$1 <= costs.length <= 10^5$
$1 <= costs[i] <= 10^5$
$1 <= k, candidates <= costs.length$

優先隊列#

class Solution {
public:
    long long totalCost(vector<int>& costs, int k, int candidates) {
        int n = costs.size();

        // 前 candidates 與後 candidates 相加大於 costs 數組的大小
        // 則說明只需要 costs 中前 k 個最小的員工代價即可
        if (candidates * 2 >= n) {
            sort(costs.begin(), costs.end());
            return accumulate(costs.begin(), costs.begin() + k, 0LL);
        }

        using PII = pair<int, int>;
        auto cmp = [&](const PII &p1, const PII &p2) {
            return p1.first == p2.first ?
                p1.second > p2.second : p1.first > p2.first;
        };
        priority_queue<PII, vector<PII>, decltype(cmp)> heap(cmp);
        // priority_queue<PII, vector<PII>, greater<PII>> heap;

        // 兩個區間，前 condidates 與後 condidates
        // left1, ..., right1, ..., left2, ..., right2
        int left1 = 0, right1 = candidates - 1;
        int left2 = n - candidates, right2 = n - 1;
        for (int i = left1; i <= right1; ++i) {
            heap.push({costs[i], i});
        }
        for (int i = left2; i <= right2; ++i) {
            heap.push({costs[i], i});
        }

        long long res = 0;
        while (k--) {
            auto t = heap.top();
            heap.pop();
            res += t.first;

            // 兩個區間存在覆蓋
            if (right1 + 1 == left2) {
                continue;
            }

            if (t.second <= right1) {
                ++right1;
                heap.push({costs[right1], right1});
            } else {
                --left2;
                heap.push({costs[left2], left2});
            }
        }
        return res;
    }
};

2462. 雇佣 K 位工人的總代價

2462. 雇佣 K 位工人的總代價#

優先隊列#

2462. 雇佣 K 位工人的總代價 #