目标
给你一个下标从 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 <= 10^5
- 1 <= costs[i] <= 10^5
- 1 <= k, candidates <= costs.length
思路
给我们一个数组 costs 和一个整数 candidates，每次从数组的前candidates与后candidates中选取值最小的元素，如果最小值相等则取下标最小的，costs中的每个值只能被选一次，求选出的k个元素和。
很直接的想法是使用优先队列保存这些元素，考虑到相同值需要取下标最小的，于是还要保存下标信息。循环从队列取出头元素累加到res，每取出一个元素需要从前/后补一个元素到队列中。
注意题目的数据范围，需要返回long类型。还有就是放队列以及补元素的停止条件以免重复。
代码
/**
* @date 2024-05-01 17:03
*/
public class TotalCost2462 {
public long totalCost(int[] costs, int k, int candidates) {
PriorityQueue<Integer[]> q = new PriorityQueue<>((x, y) -> {
int compare = x[0] - y[0];
return compare == 0 ? x[1] - y[1] : compare;
});
int n = costs.length;
int leftPos;
int rightPos = n - 1;
for (leftPos = 0; leftPos < candidates; leftPos++) {
if (leftPos < rightPos) {
// rightPos 与 leftPos指向的都是需要添加的下一个位置，
q.offer(new Integer[]{costs[leftPos], leftPos});
q.offer(new Integer[]{costs[rightPos], rightPos});
rightPos--;
}
else if (leftPos == rightPos) {
// 这里已经覆盖了数组的所有元素，循环结束后leftPos > rightPos
q.offer(new Integer[]{costs[leftPos], leftPos});
}
else {
break;
}
}
// 不使用long会溢出
long res = 0;
// 选到最小之后，补够candidates
for (int i = 0; i < k && !q.isEmpty(); i++) {
Integer[] c = q.poll();
res += c[0];
// 如果已经覆盖了所有元素就无需再添加了
if (leftPos > rightPos) {
continue;
}
if (c[1] < leftPos) {
q.offer(new Integer[]{costs[leftPos], leftPos});
leftPos++;
} else if (c[1] > rightPos) {
q.offer(new Integer[]{costs[rightPos], rightPos});
rightPos--;
}
}
return res;
}
}性能
勉强通过，官网也是这个思路，耗时82ms。有网友的解法是分别使用了两个优先队列，只需要22ms。
