Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

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