首页 > 文章列表 > 在C++中,找到使数组所有元素相等所需的操作次数

在C++中,找到使数组所有元素相等所需的操作次数

操作次数 C编程 数组相等
156 2023-09-18

在这个问题中,我们得到一个大小为 n 的数组 arr。我们的任务是找到使所有数组元素相等所需的操作数

该操作定义为从权重最大的元素到所有元素的相等权重分布数组的元素

如果不可能使数组元素相等,则打印 -1。

让我们举个例子来理解这个问题,

Input : arr[] = {7, 3, 3, 3}
Output : 3

说明

分配后的数组为 {4, 4, 4, 4}

解决方法

解决该问题的一个简单方法是找到数组的最大值。然后使用这个最大值检查数组的所有元素是否相等,并且该值是否等于数组的最大值减去n(或其倍数)。如果是,则返回n,如果否,则返回-1(表示不可能)。

示例

我们举个例子来理解一下问题

#include<bits/stdc++.h>
using namespace std;
int findOperationCount(int arr[],int n){
   int j = 0, operations = 0;
   int maxVal = arr[0];
   int minVal = arr[0];
   int maxValInd = 0;
   for (int i = 1; i < n; i++){
      if(arr[i] > maxVal){
         maxVal = arr[i];
         maxValInd = i;
      }
      if(arr[i] < minVal){
         minVal = arr[i];
      }
   }
   for (int i =0;i<n;i++){
      if (arr[i] != maxVal && arr[i] <= minVal && arr[i] != 0){
         arr[j] += 1;
         arr[maxValInd] -= 1;
         maxVal -= 1;
         operations += 1;
         j += 1;
      }
      else if (arr[i] != 0){
         j += 1;
      }
   }
   for (int i = 0; i < n; i++){
      if (arr[i] != maxVal){
         operations = -1;
         break;
      }
   }
   return operations;
}
int main(){
   int arr[] = {4, 4, 8, 4};
   int n = sizeof(arr)/sizeof(arr[0]);
   cout<<"The number of operations required to make all array  elements Equal is "<<findOperationCount(arr, n);
   return 0;
}

输出

The number of operations required to make all array elements Equal is 3