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

解题思路

左右两端candidates是两个窗口，可能有以下情况：

1. 两个窗口重叠或相接(2*candidates>=costs.size())：从两个窗口取最小值相当于从costs中取最小值，因为两个窗口包含了costs的所有元素。问题简化为对costs的前k个元素求和。
2. 两个窗口不相交，中间有未被考虑的自由数(2*canditates<costs.size())：这时不能直接选择costs的k个最小值了，比如最小值正好在中间，而查询次数k很小，窗口还不能把最小值考虑在内，最终结果会把更小的元素漏掉。注意这里k的大小影响了问题的性质，每次取出窗口的最小值后由中间自由数补充到窗口：如果k很大，所有自由数都进入窗口比较还是相当于对整个costs数组求k个最小值；如果k较小，只能按部就班对当前两个窗口的元素比较了。

所有自由数参与比较的临界条件是：costs.size()-2*candidates<k，costs.size()-2*candidates表示初始状态两个窗口没有覆盖的自由数个数，每从窗口查询一次后将一个自由数放进两个窗口之一，查询次数大于自由数的个数时，所有的自由数就都能参与比较了。注意不能是等于，等于只能保证所有自由数进入窗口，但是最后一个自由数并不参与比较，如果它是最小值还是不会被选到。

综合以上分析，if(2*candidates+k>cost.size())，直接对costs升序排列，选前k个元素求和；否则创建两个升序的优先级队列q1和q2，分别代表前后两个窗口，每次比较q1和q2队首元素取较小值，如果从q1取出，则从自由数最左端取元素补充到q1，如果从q2取出，则从自由数最右端取元素补充到q2。

代码

class Solution {
public:
   long long totalCost(vector<int>& costs, int k, int candidates) {
       long long sum = 0;
       if(2*candidates+k>costs.size()){
           sort(costs.begin(),costs.end());
           for(int i=0;i<k;++i){
               sum+=costs[i];
           }
       }
       else{
           priority_queue<int,vector<int>,greater<int>> q1;
           priority_queue<int,vector<int>,greater<int>> q2;
           int left_index = candidates;
           int right_index = costs.size()-1-candidates;
           for(int i=0;i<candidates;++i){
               q1.push(costs[i]);
               q2.push(costs[costs.size()-i-1]);
           }
           for(int i=0;i<k;++i){
               if(q1.top()<=q2.top()){
                   sum+=q1.top();
                   q1.pop();
                   q1.push(costs[left_index++]);
               }
               else{
                   sum+=q2.top();
                   q2.pop();
                   q2.push(costs[right_index--]);
               }
           }
       }
       return sum;
   }
};