Total cost to hire k workers

November 21, 2022


Problem URL: Total cost to hire k workers

We will sort the workers by their quality and iterate through them. For each worker, we will add their wage to the heap and remove the highest wage if the heap size is greater than k. We will also keep track of the sum of the wages in the heap. If the heap size is equal to k, we will calculate the total cost and update the minimum cost.

class Solution:
    def totalCost(self, costs: List[int], k: int, candidates: int) -> int:
        costs = collections.deque(costs)
        total, first, last = 0, [], []
        for _ in range(k):
            while len(first) < candidates:
                heapq.heappush(first, costs.popleft() if costs else math.inf)
            while len(last) < candidates:
                heapq.heappush(last, costs.pop() if costs else math.inf)
            total += heapq.heappop(first) if first[0] <= last[0] else heapq.heappop(last)
        return total

Time complexity: O(nlog(n))
Space complexity: O(n)