Talentd
FleetCode

1551. Minimum Operations to Make Array Equal

Medium82.8% acceptance2 companies

You have an array arr of length n where arr[i] = (2 * i) + 1 for all valid values of i (i.e., 0 <= i < n).

In one operation, you can select two indices x and y where 0 <= x, y < n and subtract 1 from arr[x] and add 1 to arr[y] (i.e., perform arr[x] -=1 and arr[y] += 1). The goal is to make all the elements of the array equal. It is guaranteed that all the elements of the array can be made equal using some operations.

Given an integer n, the length of the array, return the minimum number of operations needed to make all the elements of arr equal.

Example 1:

Input: n = 3
Output: 2
Explanation: arr = [1, 3, 5]
First operation choose x = 2 and y = 0, this leads arr to be [2, 3, 4]
In the second operation choose x = 2 and y = 0 again, thus arr = [3, 3, 3].

Example 2:

Input: n = 6
Output: 9

Constraints:

  • 1 <= n <= 104

Asked at 2 companies

Topics

Hints open one at a time

Hint 1
Build the array arr using the given formula, define target = sum(arr) / n
Hint 2
What is the number of operations needed to convert arr so that all elements equal target ?
6:02
Still stuck?Read 3 approaches, or watch a walkthrough.
Open solution
Loading editor…
1class Solution:
2 def minOperations(self, n: int) -> int:
3
Starter codePython 3Spaces: 4
Examples
Input
3
Input
6
Your code is saved on this device.Solution