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