leetcode之使数组中所有元素相等的最小操作数（C++）

参考链接

1. https://leetcode-cn.com/problems/minimum-operations-to-make-array-equal/

题目描述

存在一个长度为 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同时另一个数减1，所以数组所有元素总和不变，数组所有元素最后都等于平均值。该等差数列的平均值等于n，接着只需要计算索引为n/2之前的数变为n需要的操作数之和，因为n/2之前的是加1操作，n/2之和的是减1操作，只需要算一种。

如果将上述过程用公式推导，可以直接得到结果$\frac{n^2}{4}$

代码

常规思路

class Solution {
public:
    int minOperations(int n) {
        int res = 0;
        for (int i = 0; i < n / 2; i ++)
        {
            res += n - (2 * i + 1);
        }
        return res;
    }
};

公式法

class Solution {
public:
    int minOperations(int n) {
        return n * n / 4;
    }
};