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

题目

存在一个长度为 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

提示：

• 1 <= n <= 10^4

解法一：平均值求差法（自己思考的解法）

解题思路：

1.求出数组中所有元素的和，因为数组是一个从1开始，长度为n的奇数等差出列。这里可以用等差数列的求和公式 (上底 + 下底) * 高 / 2 。sum = (1 + 2 * (n - 1) + 1) * n / 2 = n * n

2.求平均值，sum / n = n

3.注意到每次我们进行操作时都同时进行了「加」操作和「减」操作，这样我们只需要记录「加」操作的数量即可知道我们一共操作了多少次。

遍历数组中所有小余平均值的元素，平均值与该元素的差值即为该元素需要进行「加」操作的数量，我们只需要统计累加这些差值，就能计算出我们一共需要操作多少次。

具体实现代码如下：

    public static int minOperations(int n) {
    	//求出数组中所有元素的和。用等差数列求和公式 (首项 + 尾项) * 高 / 2
    	int sum = n * n;
    	//求出n个元素的平均值，也就是所有操作结束之后，所有元素相等的值。
    	int average = n;
    	int i = 0;
    	int operation = 0;
    	//遍历数组前半部分所有小余平均值的元素，累计需要“加”操作的数量。因为加操作和减操作是同步进行的，所以只累计“加”操作就可以。
    	while(2 * i + 1 < average){
    		operation += average - 2 * i - 1;
    		i++;
    	}
    	return operation;
    }

解法二：数学分析法

我们来分析一下。

举例一：n是奇数

n=5，arr[] = {1, 3, 5, 7, 9}，我们希望通过「加」和「减」操作，把所有数都变成相等的，这里显然要变成5（平均值是5）。那么我们怎么变呢？arr[] = {1 + 4, 3 + 2, 5, 7 - 2, 9 - 4}，所以最少操作次数 2 + 4 = 6。

n=7，arr[] = {1, 3, 5, 7, 9, 11, 13}，所有的数都要变成7（平均值是7），arr[] = {1 + 6, 3 + 4, 5 + 2, 7 , 9 - 2, 11 - 4, 13 - 6}，所以最少操作次数 2 + 4 + 6 = 12。

所以通过观察，最少操作次数的累计，每次都是加一个偶数，比如2，4，6，8。因此这个问题就变成了求前 (n - 1)/2 个正偶数的和。

然后用等差数列的求和公式，首项 2，尾项是 2 * (n - 1) / 2 = n - 1。

操作次数 = （2 + n -1） * (n - 1) / 2 / 2 = (n + 1)*(n - 1) / 4 = (n * n - 1) / 4

举例二：n是偶数

n = 6，arr[] = {1, 3, 5, 7, 9}，所有的数都要变成6（平均值是6），arr[] = {1 + 5, 3 + 3, 5 + 1, 7 - 1, 9 - 3, 11 - 5}，所以最少操作次数 1 + 3 + 5 = 9。

所以通过观察（可以自己再去多测试一些数据），最少操作次数的累计，每次都是加一个奇数，比如1，3，5，7。因此这个问题就变成了求前 n/2 个正奇数的和。

然后用等差数列的求和公式，首项 1，尾项是 n / 2 * 2 - 1 = n - 1。

操作次数 = (1 + n -1) * n /2 / 2 = n * n / 4

总结一下：

n是奇数：(n + 1)*(n - 1) / 4 = (n * n - 1) / 4

n是偶数：n * n / 4

统一成一个式子：(n * n - (n & 1)) >>2

n & 1表示n和1做与运算，相当于n对2取余数，简单的说就是n是奇数的时候取1，n是偶数的时候取2。

>>2表示除于4

    public static int minOperations(int n) {
    	return (n * n - (n & 1)) >>2;
    }