题目描述 :
给你一个下标从 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 。
题解:
雇佣过程可以用两个最小堆来模拟,一个负责维护 costs剩余数字的最前面 candidates 个数的最小值,另一个负责维护 costs 剩余数字的最后面 candidates 个数的最小值。
具体算法如下:
- 设 cost 的长度为 n。如果 candidates⋅2+k>n,我们一定可以选到 cost 中最小的 k 个数,所以直接返回 cost 的前 k 小之和,因此无需判断是否有重合。
- 初始化答案 ans=0。初始化最小堆 pre 为 costs 最前面的 candidates 个数,初始化最小堆 suf 为 costs 最后面的 candidates 个数。初始化下标 i=candidates, j=n−1−candidates。
- 循环 k 次。每次循环,如果 pre 的堆顶小于等于 suf 的堆顶,则弹出 pre 的堆顶,加入答案,然后把 costs 加入 pre,i 增加 1;如果 suf\ 的堆顶小于 pre 的堆顶,则弹出 suf 的堆顶,加入答案,然后把 costs 加入 suf,j 减少 1。
- 返回答案。
代码:
class Solution {
public long totalCost(int[] costs, int k, int candidates) {
long ans = 0;
int n = costs.length;
if (candidates * 2 + k > n) {
Arrays.sort(costs);
for (int i = 0; i < k; i++) {
ans += costs[i];
}
return ans;
}
PriorityQueue<Integer> pre = new PriorityQueue<>();
PriorityQueue<Integer> suf = new PriorityQueue<>();
for (int i = 0; i < candidates; i++) {
pre.offer(costs[i]);
suf.offer(costs[n - 1 - i]);
}
int i = candidates;
int j = n - 1 - candidates;
while (k-- > 0) {
if (pre.peek()<=suf.peek()) {
ans+= pre.poll();
pre.offer(costs[i++]);
} else {
ans+=suf.poll();
suf.offer(costs[j--]);
}
}
return ans;
}
}