前缀和：leetcode2602数组元素全部相等的最少操作次数

最新推荐文章于 2024-11-10 14:06:08 发布

坠金

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量150

点赞数

分类专栏：技术面算法文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/qq_41413211/article/details/130078363

版权

技术面算法专栏收录该内容

71 篇文章 1 订阅 ¥9.90 ¥99.00

订阅专栏

该博客讨论了如何利用前缀和优化算法，以最少的操作次数使数组所有元素相等。针对LeetCode的2602题目，博主指出暴力枚举方法会超时，并详细解释了如何运用前缀和来加速求解过程，避免了二重循环导致的时间复杂度问题。

摘要由CSDN通过智能技术生成

2602. 使数组元素全部相等的最少操作次数 - 力扣（LeetCode）

暴力枚举（该方法超时）

这道题最直观的做法

每一轮query中，都直接计算nums[i]和quries[i]的差值之和

class Solution:
    def minOperations(self, nums: List[int], queries: List[int]) -> List[int]:
        res = []
        for q in queries:
            cnt = 0
            for num in nums:
                cnt += abs(q - num)
            res.append(cnt)
        return res

可以看到耗时的地方在于二重循环。因为需要求和，所以在二重循环中运算。看到求和，就要马上想到前缀和，更进一步的，我们应该如何用前缀和加速求解？