1. 程式人生 > 其它 >陣列中所有元素相等的最小運算元--java

陣列中所有元素相等的最小運算元--java

存在一個長度為 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)
連結:https://leetcode-cn.com/problems/minimum-operations-to-make-array-equal
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

class Solution {
    //  0   1   2   3   4   5   6  
    //  1   3   5   7   9   11  13
    public int minOperations(int n) {
        int[] arr = new int[n];
        for(int i=0;i<n;i++){//遍歷生成陣列
            arr[i]=(2*i)+1;
        }
        int count=0;//計數器
        int left=0,right=n-1;//頭尾指標
        while(true){//根據規律看出頭尾相加除二再減一就是需要加的次數
            if(arr[left]!=arr[right]&&left<right){
                int num = (arr[left]+arr[right])/2;
                num = num - arr[left];
                count = count + num;
                left++;
                right--;
            }
            if(left==right||left>right){//特殊情況退出
                break;
            }
        }
        return count;
    }
}