Number of operations required to make all array elements Equal in Python

We have given an array of elements, and we have to make them all equal by incrementing the elements by 1. We are allowed to increment n - 1 element at each step. Our goal is to calculate the total number of operations required to make all the array elements equal.

For example, if you take the list [1, 2, 3], it took three operations to make all the elements equal. One solution to the problem is. Find the most significant number at each step and increment the rest of the elements by 1. Let's write the code.

Example

 Live Demo

def main():
   # intializing the array
   arr = [1, 2, 3]
   # initializing operations count to 0
   no_of_operations = 0
   flag = 0
   # performing the operations on array to make them equal
   while not are_equal(arr):
      flag = 1
      # finding the maximum from the list
      maximum = max(arr)
      # incrementing all the elements except maximum
      for i in range(len(arr)):
         if arr[i] != maximum:
            arr[i] += 1
      # incrementing the operations count by 1
      no_of_operations += 1
   print(no_of_operations) if flag == 0 else print(no_of_operations + 1)
# checking whether all the elements are equal or not
def are_equal(arr):
   global no_of_operations
   for i in range(len(arr) - 1):
      if arr[i] != arr[i + 1]:
         return False
   return True
if __name__ == '__main__':
   main()

Output

If you run the above program, you will get the following results.

3

The above method takes more time to compute for large arrays. We can find the number of operations by finding the sum and the smallest element of the array.

• Find the sum of the array.
• Find the smallest of all elements in the array.
• Print the value get from the expression sum - (length - smallest).

Example

See the code below.

 Live Demo

# initializing an array
arr = [1, 2, 3]
# length
length = len(arr)
# sum of element fo the array
elements_sum = sum(arr)
# smallest among all the elements
smallest = min(arr)
# calculating the number of operations
print(elements_sum - (length * smallest))

Output

If you run the above code, you will get the following results.

3

Conclusion

The second method that we discussed is easier and takes less time compared to the first method. If you have any doubts in the tutorial, mention them in the comment section.

Hafeezul Kareem

Published on 02-Jan-2020 06:08:17

• Related Questions & Answers
• Minimum operations required to make all the array elements equal in C++
• Minimum number of operations required to delete all elements of the array using C++.
• Minimum number of given operations required to make two strings equal using C++.
• Minimum number of operations on an array to make all elements 0 using C++.
• Minimum operations of given type to make all elements of a matrix equal in C++
• Minimum number of moves to make all elements equal using C++.
• Minimum delete operations to make all elements of array same in C++.
• Program to find minimum number of operations required to make one number to another in Python
• Program to find minimum operations to make array equal using Python
• Minimum operation to make all elements equal in array in C++
• Program to count minimum number of operations required to make numbers non coprime in Python?
• Program to find minimum number of operations required to make lists strictly Increasing in python
• Minimum operations required to set all elements of binary matrix in C++
• Program to find minimum number of operations required to make one string substring of other in Python
• Find the minimum number of preprocess moves required to make two strings equal in Python