1551 使数组中所有元素相等的最小操作数

1. 问题描述：

存在一个长度为 n 的数组 arr ，其中 arr[i] = (2 * i) + 1 （ 0 <= i < n ）。
一次操作中，你可以选出两个下标，记作 x 和 y （ 0 <= x, y < n ）并使 arr[x] 减去 1 、arr[y] 加上 1 （即 arr[x] -=1 且 arr[y] += 1 ）。最终的目标是使数组中的所有元素都 相等 。题目测试用例将会 保证 ：在执行若干步操作后，数组中的所有元素最终可以全部相等。
给你一个整数 n，即数组的长度。请你返回使数组 arr 中所有元素相等所需的最小操作数 

示例 1：

输入：n = 3
输出：2
解释：arr = [1, 3, 5]
第一次操作选出 x = 2 和 y = 0，使数组变为 [2, 3, 4]
第二次操作继续选出 x = 2 和 y = 0，数组将会变成 [3, 3, 3]

示例 2：

输入：n = 6
输出：9
来源：力扣（LeetCode）

提示：

• 1 <= n <= 10^4

链接：https://leetcode-cn.com/problems/minimum-operations-to-make-array-equal

2. 思路分析：

因为每一次都是选择两个元素进行加1减1的操作，所以经过若干次操作之后元素总和还是相等的，并且最终所有元素是相等的，所以这个元素肯定是所有元素的平均值，我们只需要根据数组长度为n进行遍历得到2 * i + 1的元素判断当前元素是否大于了平均值即可

3. 代码如下：

class Solution:
    def minOperations(self, n: int) -> int:
        avg, res = n, 0
        for i in range(0, n):
            if 2 * i + 1 > avg: res += 2 * i + 1 - n
        return res

python官方代码：

class Solution:
    def minOperations(self, n: int) -> int:
        return sum(x - n for i in range(n) if (x := 2 * i + 1) > n)