2462. 雇佣 K 位工人的总代价
2462. 雇佣 K 位工人的总代价 (Medium)
给你一个下标从 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⁵1 <= costs[i] <= 10⁵1 <= k, candidates <= costs.length
分析
维护左右两个小顶堆即可。
func totalCost(costs []int, k int, candidates int) int64 {
i, j := 0, len(costs)-1
left, right := &Heap{}, &Heap{}
var res int64
for ; k > 0; k-- {
for left.Len() < candidates && i <= j {
left.push(costs[i])
i++
}
for right.Len() < candidates && i <= j {
right.push(costs[j])
j--
}
if right.Len() == 0 || left.Len() > 0 && left.peek() <= right.peek() {
res += int64(left.pop())
} else {
res += int64(right.pop())
}
}
return res
}
type Heap struct {
sort.IntSlice
}
func (h *Heap) Push(x any) { h.IntSlice = append(h.IntSlice, x.(int)) }
func (h *Heap) Pop() any {
n := h.Len()
x := h.IntSlice[n-1]
h.IntSlice = h.IntSlice[:n-1]
return x
}
func (h *Heap) push(x int) { heap.Push(h, x) }
func (h *Heap) pop() int { return heap.Pop(h).(int) }
func (h *Heap) peek() int { return h.IntSlice[0] }