Equalize an array using array elements only in C++

In this problem, we are given an array of n elements. Our task is to create a program to count the number of operations to equalize an array using elements only.

We need to count the number of adding or subtraction operations that will be performed to make all the elements of the array equal.

Let’s take an example to understand the problem,

Input: arr[] = {4, 0, 3, 1, 2}

Output: 3

Explanation: 

The equal value will be 2.

The overall sum will be the same. We will be taking 1 from value at arr[3] and then adding it to value at arr[1].

Then we will take 2 from value at arr[0] and add it to value at arr[1].

Solution Approach:

A simple solution to the problem is by finding the element from the array to be treated as the elements which will be the equal element for the array.

We will check if the operation is possible by finding the average, if it is integer then equalisation is possible otherwise not.

If equalisation is possible, we will find the count the number of operations that are required and then return it. The count of operation is equal to half of the sum of absolute difference of all numbers with the average.

Algorithm :

Step 1: find the average of all elements of the array.

Step 2: if average is not an integer, return -1, indicating equalisation is not possible.

Step 3: else, find the absolute difference between all elements and the average.

Step 4: Return half of the average value.

Program to illustrate the working of our solution,

Example

Live Demo

#include <bits/stdc++.h>
using namespace std;

int calcEqualisedOperations(int arr[], int n) {
   
   int sum = 0, average, operations = 0;
   for (int i = 0; i < n; i++)
      sum += arr[i];
   if (sum % n != 0)
      return -1;
   average = sum/n;
   for (int i = 0; i < n; i++)
      operations += ( abs(arr[i] - average) / 2 );

   return operations;
}

int main() {

   int arr[] = { 5, 3, 2, 6 };
   int n = sizeof(arr) / sizeof(arr[0]);
   cout<<"Operations required to equalize an array using array elements is "<<calcEqualisedOperations(arr, n);
   return 0;
}

Output −

Operations required to equalize an array using array elements is 2

sudhir sharma

Updated on 22-Jan-2021 12:55:44

• Related Questions & Answers
• Inserting elements in an array using C Language
• Finding the only unique string in an array using JavaScript
• Find the only different element in an array using C++
• Rank of All Elements in an Array using C++
• Find original array from encrypted array (An array of sums of other elements) using C++.
• Deep count of elements of an array using JavaScript
• Display only an element found in an array in MongoDB?
• Can you assign an Array of 100 elements to an array of 10 elements in Java?
• Update elements inside an array in MongoDB?
• Rearranging elements of an array in JavaScript
• C++ Program to Access Elements of an Array Using Pointer
• Sorting only a part of an array JavaScript
• Counting elements of an array using a recursive function in JS?
• Repeating only even numbers inside an array in JavaScript
• Finding the only out of sequence number from an array using JavaScript