LC-2602 Medium LeetCode

2602. Minimum Operations to Make All Array Elements Equal

Read the full problem statement on LeetCode.

Difficulty: medium Acceptance: 37% Topics: Array, Binary Search, Sorting, Prefix Sum

Reading material

Arrays & Hashing 12 min Binary Search 12 min Sorting 13 min

Reference solution (spoiler · python)

# Time:  O(nlogn + qlogn)
# Space: O(n)

# sort, binary search, prefix sum
class Solution(object):
    def minOperations(self, nums, queries):
        """
        :type nums: List[int]
        :type queries: List[int]
        :rtype: List[int]
        """
        nums.sort()
        prefix = [0]*(len(nums)+1)
        for i in xrange(len(nums)):
            prefix[i+1] = prefix[i]+nums[i]
        result = [0]*len(queries)
        for i, q in enumerate(queries):
            j = bisect.bisect_left(nums, q)
            result[i] = (q*j-prefix[j])+((prefix[-1]-prefix[j])-q*(len(nums)-j))
        return result

Solution from kamyu104/LeetCode-Solutions · MIT