在 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
广告