Pythonですべての配列要素を等しくするために必要な操作の数

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

上記の方法では、大きな配列の計算に時間がかかります。配列の合計と最小要素を見つけることで、演算の数を見つけることができます。

• 配列の合計を求めます。
• 配列内のすべての要素の中で最小のものを見つけます。
• 式の合計から取得した値を出力します-（長さ-最小） 。

例

以下のコードを参照してください。

# 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