存在一个长度为 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
贪心法,相当于调整数组中的数,使每个数都等于n,利用求和公式即可:
class Solution {
public:int minOperations(int n) {int res = 0;if (n & 1){// 拿5举例,数组中的3需要2步到5,1需要4步到5,即求2 + 4的值// 相当于以2为首数字,公差为2的等差数列求和,数列的项数为n / 2res = (2 + 2 * (n / 2)) * (n / 2) / 2;}else{// 拿4距离,数组中的3需要1步到4,1需要3步到4,即求1 + 3的值// 相当于以1位首数字,公差为2的等差数列求和,数列的项数为n / 2// res = (1 + 2 * (n / 2) - 1) * (n / 2) / 2;res = n * n / 4;}// 实际上,当n为奇数时,res的结果可直接用偶数的公式n * n / 4算出// 而不能用(1 + 2 * (n / 2) - 1) * (n / 2) / 2;计算出,因为n/2会舍弃小数// 因此本题可以直接返回n * n / 4return res;}
};
此算法时间复杂度为O(1),空间复杂度为O(1)。