在 C++ 中查找使所有数组元素相等所需的运算次数


在这个问题中,我们给定一个大小为 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

更新于:2022年1月24日

320 次浏览

开始您的 职业生涯

完成课程后获得认证

开始
广告