存在一個長度為 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)。