2462. 雇佣 K 位工人的总代价

给你一个下标从 0 开始的整数数组 costs ，其中 costs[i] 是雇佣第 i 位工人的代价。

同时给你两个整数 k 和 candidates 。我们想根据以下规则恰好雇佣 k 位工人：

• 总共进行 k 轮雇佣，且每一轮恰好雇佣一位工人。
• 在每一轮雇佣中，从最前面 candidates 和最后面 candidates 人中选出代价最小的一位工人，如果有多位代价相同且最小的工人，选择下标更小的一位工人。
  • 比方说，costs = [3,2,7,7,1,2] 且 candidates = 2 ，第一轮雇佣中，我们选择第 4 位工人，因为他的代价最小 [_3,2_,7,7,_**1**,2_] 。
  • 第二轮雇佣，我们选择第 1 位工人，因为他们的代价与第 4 位工人一样都是最小代价，而且下标更小，[_3,**2**_,7,_7,2_] 。注意每一轮雇佣后，剩余工人的下标可能会发生变化。
• 如果剩余员工数目不足 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 <= 105
• 1 <= costs[i] <= 105
• 1 <= k, candidates <= costs.length

思路（双指针 + 优先队列）

这里分为两种情况，首先第一种是在左边和右边选取最小值的时候，两部分出现了重合

那么这个时候左边前 candidates 和 右边前candidates 其实就已经包括了全部的数据，我们就只需要取出整个数组的最小值即可，所以可以得到以下结论：当两部分重合时，我们所求的，就是数组从小到大排序后的前 k 个值。

第二种情况就是没有重合，那么这时候就依次看左右两边，每次存最小的。具体做法为：创建两个小根堆，分别存储左边前 candidates 个值和右边 的candidates 个值，每次我们取出对头，将答案加上最小的那个，并把那一边的队友弹出，然后向后添加一个元素，一直取 k 个。

代码

class Solution {
public:
    long long totalCost(vector<int>& costs, int k, int candidates) {
        int n = costs.size();
        long long res = 0;
        priority_queue<int, vector<int>, greater<int>> l, r;

        int a = candidates, b = costs.size() - candidates - 1;
        if ( candidates * 2 + k > n ) {
            sort(costs.begin(), costs.end());
            for ( int i = 0; i < k; i ++ ) res += costs[i];
            return res;
        }

        for ( int i = 0; i < candidates; i ++ ) {
            l.push(costs[i]);
            r.push(costs[costs.size() - 1 - i]);
        }

        while ( k -- ) {
            int lt = l.top(), rt = r.top();

            if ( lt <= rt ) {
                res += lt;
                l.pop();
                l.push(costs[a ++]);
            } else {
                res += rt;
                r.pop();
                r.push(costs[b --]);
            }
        }

        return res;
    }
}; 
`