使所有数组元素在Python中相等所需的操作数

我们给定了一个元素数组，我们必须通过将元素递增1来使它们全部相等。我们被允许在每一步中递增n-1元素。我们的目标是计算使所有数组元素相等所需的操作总数。

例如，如果采用列表[1、2、3]，则需要进行三个操作才能使所有元素相等。解决该问题的一种方法是。在每个步骤中找到最重要的数字，并将其余元素加1。让我们编写代码。

示例

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()

输出结果

如果运行上面的程序，您将得到以下结果。

3

上面的方法需要更多的时间来计算大型数组。我们可以通过求和和数组的最小元素来找到操作数。

• 查找数组的总和。

• 查找数组中所有元素中的最小元素。

• 打印从表达式sum- (length-smallest)得到的值。

示例

请参见下面的代码。

# 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))

输出结果

如果运行上面的代码，您将得到以下结果。

3

结论

与第一种方法相比，我们讨论的第二种方法更容易且耗时更少。如果您对本教程有任何疑问，请在评论部分中提及。