Divide array into two sub-arrays such that their averages are equal
Last Updated :
12 Jul, 2025
Given an integer array, the task is to divide an integer array into two sub-arrays to make their averages equal if possible.
Examples :
Input : arr[] = {1, 5, 7, 2, 0};
Output : (0 1) and (2 4)
Subarrays arr[0..1] and arr[2..4] have
same average.
Input : arr[] = {4, 3, 5, 9, 11};
Output : Not possible
Asked in Microsoft
A Naive Approach is to run two loops and find subarrays whose averages are equal.
Implementation:
C++
// Simple C++ program to find subarrays
// whose averages are equal
#include<bits/stdc++.h>
using namespace std;
// Finding two subarrays
// with equal average.
void findSubarrays(int arr[], int n)
{
bool found = false;
int lsum = 0;
for (int i = 0; i < n - 1; i++)
{
lsum += arr[i];
int rsum = 0;
for (int j = i + 1; j < n; j++)
rsum += arr[j];
// If averages of arr[0...i] and
// arr[i+1..n-1] are same. To avoid
// floating point problems we compare
// "lsum*(n-i+1)" and "rsum*(i+1)"
// instead of "lsum/(i+1)" and
// "rsum/(n-i+1)"
if (lsum * (n - i - 1) ==
rsum * (i + 1))
{
printf("From (%d %d) to (%d %d)\n",
0, i, i + 1, n - 1);
found = true;
}
}
// If no subarrays found
if (found == false)
cout << "Subarrays not found"
<< endl;
}
// Driver code
int main()
{
int arr[] = {1, 5, 7, 2, 0};
int n = sizeof(arr) / sizeof(arr[0]);
findSubarrays(arr, n);
return 0;
}
// Simple Java program to find subarrays
// whose averages are equal
public class GFG {
// Finding two subarrays
// with equal average.
static void findSubarrays(int[] arr, int n)
{
boolean found = false;
int lsum = 0;
for (int i = 0; i < n - 1; i++)
{
lsum += arr[i];
int rsum = 0;
for (int j = i + 1; j < n; j++)
rsum += arr[j];
// If averages of arr[0...i] and
// arr[i+1..n-1] are same. To avoid
// floating point problems we compare
// "lsum*(n-i+1)" and "rsum*(i+1)"
// instead of "lsum/(i+1)" and
// "rsum/(n-i+1)"
if (lsum * (n - i - 1) ==
rsum * (i + 1))
{
System.out.println("From (0 " + i
+ ") to (" +(i + 1) + " "
+ (n - 1)+ ")");
found = true;
}
}
// If no subarrays found
if (found == false)
System.out.println( "Subarrays not "
+ "found");
}
// Driver code
static public void main (String[] args)
{
int[] arr = {1, 5, 7, 2, 0};
int n = arr.length;
findSubarrays(arr, n);
}
}
// This code is contributed by Mukul Singh.
# Simple Python 3 program to find subarrays
# whose averages are equal
# Finding two subarrays with equal average.
def findSubarrays(arr, n):
found = False
lsum = 0
for i in range(n - 1):
lsum += arr[i]
rsum = 0
for j in range(i + 1, n):
rsum += arr[j]
# If averages of arr[0...i] and
# arr[i+1..n-1] are same. To avoid
# floating point problems we compare
# "lsum*(n-i+1)" and "rsum*(i+1)"
# instead of "lsum/(i+1)" and
# "rsum/(n-i+1)"
if (lsum * (n - i - 1) == rsum * (i + 1)):
print("From", "(", 0, i, ")",
"to", "(", i + 1, n - 1, ")")
found = True
# If no subarrays found
if (found == False):
print("Subarrays not found")
# Driver code
if __name__ == "__main__":
arr = [1, 5, 7, 2, 0]
n = len(arr)
findSubarrays(arr, n)
# This code is contributed by ita_c
// Simple C# program to find subarrays
// whose averages are equal
using System;
public class GFG {
// Finding two subarrays
// with equal average.
static void findSubarrays(int []arr, int n)
{
bool found = false;
int lsum = 0;
for (int i = 0; i < n - 1; i++)
{
lsum += arr[i];
int rsum = 0;
for (int j = i + 1; j < n; j++)
rsum += arr[j];
// If averages of arr[0...i] and
// arr[i+1..n-1] are same. To avoid
// floating point problems we compare
// "lsum*(n-i+1)" and "rsum*(i+1)"
// instead of "lsum/(i+1)" and
// "rsum/(n-i+1)"
if (lsum * (n - i - 1) ==
rsum * (i + 1))
{
Console.WriteLine("From ( 0 " + i
+ ") to(" + (i + 1) + " "
+ (n - 1) + ")");
found = true;
}
}
// If no subarrays found
if (found == false)
Console.WriteLine( "Subarrays not "
+ "found");
}
// Driver code
static public void Main ()
{
int []arr = {1, 5, 7, 2, 0};
int n = arr.Length;
findSubarrays(arr, n);
}
}
// This code is contributed by anuj_67.
<?php
// Simple PHP program to find subarrays
// whose averages are equal
// Finding two subarrays
// with equal average.
function findSubarrays( $arr, $n)
{
$found = false;
$lsum = 0;
for ( $i = 0; $i < $n - 1; $i++)
{
$lsum += $arr[$i];
$rsum = 0;
for ( $j = $i + 1; $j < $n; $j++)
$rsum += $arr[$j];
// If averages of arr[0...i] and
// arr[i+1..n-1] are same. To avoid
// floating point problems we compare
// "lsum*(n-i+1)" and "rsum*(i+1)"
// instead of "lsum/(i+1)" and "rsum/(n-i+1)"
if ($lsum * ($n - $i - 1) ==
$rsum * ($i + 1))
{
echo "From ( 0 ", $i," )".
" to (", $i + 1," ", $n - 1,")\n";
$found = true;
}
}
// If no subarrays found
if ($found == false)
echo "Subarrays not found" ;
}
// Driver code
$arr = array(1, 5, 7, 2, 0);
$n = count($arr);
findSubarrays($arr, $n);
// This code is contributed by vt_m
?>
<script>
// Simple Javascript program to find subarrays
// whose averages are equal
// Finding two subarrays
// with equal average.
function findSubarrays(arr,n)
{
let found = false;
let lsum = 0;
for (let i = 0; i < n - 1; i++)
{
lsum += arr[i];
let rsum = 0;
for (let j = i + 1; j < n; j++)
rsum += arr[j];
// If averages of arr[0...i] and
// arr[i+1..n-1] are same. To avoid
// floating point problems we compare
// "lsum*(n-i+1)" and "rsum*(i+1)"
// instead of "lsum/(i+1)" and
// "rsum/(n-i+1)"
if (lsum * (n - i - 1) ==
rsum * (i + 1))
{
document.write("From (0 " + i
+ ") to (" +(i + 1) + " "
+ (n - 1)+ ")");
found = true;
}
}
// If no subarrays found
if (found == false)
document.write( "Subarrays not "
+ "found");
}
// Driver code
let arr=[1, 5, 7, 2, 0];
let n = arr.length;
findSubarrays(arr, n);
// This code is contributed by avanitrachhadiya2155
</script>
OutputFrom (0 1) to (2 4)
Time complexity : O(n2)
Auxiliary Space : O(1)
An Efficient solution is to find sum of array elements. Initialize leftsum as zero. Run a loop and find leftsum by adding elements of array. For rightsum, we subtract leftsum from total sum then we find rightsum and find average of leftsum and rightsum as according to their index.
1) Compute sum of all array elements. Let this
sum be "sum"
2) Initialize leftsum = 0.
3) Run a loop for i=0 to n-1.
a) leftsum = leftsum + arr[i]
b) rightsum = sum - leftsum
c) If average of left and right are same,
print current index as output.
Below is the implementation for above approach:
C++
// Efficient C++ program for
// dividing array to make
// average equal
#include<bits/stdc++.h>
using namespace std;
void findSubarrays(int arr[], int n)
{
// Find array sum
int sum = 0;
for (int i = 0; i < n; i++)
sum += arr[i];
bool found = false;
int lsum = 0;
for (int i = 0; i < n - 1; i++)
{
lsum += arr[i];
int rsum = sum - lsum;
// If averages of arr[0...i]
// and arr[i+1..n-1] are same.
// To avoid floating point problems
// we compare "lsum*(n-i+1)"
// and "rsum*(i+1)" instead of
// "lsum/(i+1)" and "rsum/(n-i+1)"
if (lsum * (n - i - 1) == rsum * (i + 1))
{
printf("From (%d %d) to (%d %d)\n",
0, i, i+1, n-1);
found = true;
}
}
// If no subarrays found
if (found == false)
cout << "Subarrays not found"
<< endl;
}
// Driver code
int main()
{
int arr[] = {1, 5, 7, 2, 0};
int n = sizeof(arr) / sizeof(arr[0]);
findSubarrays(arr, n);
return 0;
}
// Efficient Java program for
// dividing array to make
// average equal
import java.util.*;
class GFG
{
static void findSubarrays(int arr[], int n)
{
// Find array sum
int sum = 0;
for (int i = 0; i < n; i++)
sum += arr[i];
boolean found = false;
int lsum = 0;
for (int i = 0; i < n - 1; i++)
{
lsum += arr[i];
int rsum = sum - lsum;
// If averages of arr[0...i]
// and arr[i+1..n-1] are same.
// To avoid floating point problems
// we compare "lsum*(n-i+1)"
// and "rsum*(i+1)" instead of
// "lsum/(i+1)" and "rsum/(n-i+1)"
if (lsum * (n - i - 1) == rsum * (i + 1))
{
System.out.printf("From (%d %d) to (%d %d)\n",
0, i, i + 1, n - 1);
found = true;
}
}
// If no subarrays found
if (found == false)
System.out.println("Subarrays not found");
}
// Driver code
static public void main ( String []arg)
{
int arr[] = {1, 5, 7, 2, 0};
int n = arr.length;
findSubarrays(arr, n);
}
}
// This code is contributed by Princi Singh
# Efficient Python program for
# dividing array to make
# average equal
def findSubarrays(arr, n):
# Find array sum
sum = 0;
for i in range(n):
sum += arr[i];
found = False;
lsum = 0;
for i in range(n - 1):
lsum += arr[i];
rsum = sum - lsum;
# If averages of arr[0...i]
# and arr[i + 1..n - 1] are same.
# To avoid floating point problems
# we compare "lsum*(n - i + 1)"
# and "rsum*(i + 1)" instead of
# "lsum / (i + 1)" and "rsum/(n - i + 1)"
if (lsum * (n - i - 1) == rsum * (i + 1)):
print("From (%d %d) to (%d %d)\n"%
(0, i, i + 1, n - 1));
found = True;
# If no subarrays found
if (found == False):
print("Subarrays not found");
# Driver code
if __name__ == '__main__':
arr = [ 1, 5, 7, 2, 0 ];
n = len(arr);
findSubarrays(arr, n);
# This code is contributed by Rajput-Ji
// Efficient C# program for
// dividing array to make
// average equal
using System;
class GFG
{
static void findSubarrays(int []arr, int n)
{
// Find array sum
int sum = 0;
for (int i = 0; i < n; i++)
sum += arr[i];
bool found = false;
int lsum = 0;
for (int i = 0; i < n - 1; i++)
{
lsum += arr[i];
int rsum = sum - lsum;
// If averages of arr[0...i]
// and arr[i+1..n-1] are same.
// To avoid floating point problems
// we compare "lsum*(n-i+1)"
// and "rsum*(i+1)" instead of
// "lsum/(i+1)" and "rsum/(n-i+1)"
if (lsum * (n - i - 1) == rsum * (i + 1))
{
Console.Write("From ({0} {1}) to ({2} {3})\n",
0, i, i + 1, n - 1);
found = true;
}
}
// If no subarrays found
if (found == false)
Console.WriteLine("Subarrays not found");
}
// Driver code
static public void Main ( String []arg)
{
int []arr = {1, 5, 7, 2, 0};
int n = arr.Length;
findSubarrays(arr, n);
}
}
// This code is contributed by Rajput-Ji
<script>
// Efficient Javascript program for
// dividing array to make
// average equal
function findSubarrays(arr,n)
{
// Find array sum
let sum = 0;
for (let i = 0; i < n; i++)
sum += arr[i];
let found = false;
let lsum = 0;
for (let i = 0; i < n - 1; i++)
{
lsum += arr[i];
let rsum = sum - lsum;
// If averages of arr[0...i]
// and arr[i+1..n-1] are same.
// To avoid floating point problems
// we compare "lsum*(n-i+1)"
// and "rsum*(i+1)" instead of
// "lsum/(i+1)" and "rsum/(n-i+1)"
if (lsum * (n - i - 1) == rsum * (i + 1))
{
document.write(
"From (0 "+i+") to ("+(i+1)+" "+(n-1)+")\n"
);
found = true;
}
}
// If no subarrays found
if (found == false)
document.write("Subarrays not found");
}
// Driver code
let arr=[1, 5, 7, 2, 0];
let n = arr.length;
findSubarrays(arr, n);
// This code is contributed by rag2127
</script>
OutputFrom (0 1) to (2 4)
Time complexity : O(n)
Auxiliary Space : O(1)
Similar Reads
Break an array into maximum number of sub-arrays such that their averages are same Given an integer array, the task is to divide the array into the maximum number of sub-arrays such that the averages of all subarrays are the same. If it is not possible to divide, then print "Not possible". Examples: Input : arr[] = {1, 5, 7, 2, 0}; Output : (0 1) (2 4) Subarrays arr[0..1] and arr[
8 min read
Generate an Array such that Average of triplet exist in given Array Given an array A[] of length N and your task is to find an array B[] of size N+2, such that Average of (B[i], B[i+1], B[i+2]) equals to A[i] for each i (1 <= i <= N) . Note: First two elements B[] must be zero. Examples:Input: N = 1, A[] = {3} Output: B[] = {0, 0, 9} Explanation: Let us unders
5 min read
Find an Array such that mean of this and given Array together equals to K Given two integers N, K, and an array arr[] consisting of positive integers. The task is to find any possible array of size N such that the mean of arr[] and the array to be found together is K. Examples: Input: arr[] = {1, 5, 6}, N = 4, K = 3Output: {1, 2, 3, 3}Explanation: Mean of {1, 5, 6} and {1
7 min read
Find an element in array such that sum of left array is equal to sum of right array Given, an array of size n. Find an element that divides the array into two sub-arrays with equal sums. Examples: Input: 1 4 2 5 0Output: 2Explanation: If 2 is the partition, subarrays are : [1, 4] and [5] Input: 2 3 4 1 4 5Output: 1Explanation: If 1 is the partition, Subarrays are : [2, 3, 4] and [4
15+ min read
Find if array can be divided into two subarrays of equal sum Given an array of integers, find if it's possible to remove exactly one integer from the array that divides the array into two subarrays with the same sum. Examples: Input: arr = [6, 2, 3, 2, 1] Output: true Explanation: On removing element 2 at index 1, the array gets divided into two subarrays [6]
9 min read
Find array elements that are greater than average Given an array of numbers, print all those elements that are greater than average. Examples: Input : 5, 4, 6, 9, 10 Output : 9 10 Explanation: avg = 5 + 4 + 6 + 9 + 10 / 5; avg = 34 / 5 avg = 6.8 Elements greater than 6.8 are 9 and 10 Input : 1, 2, 4, 0, 5 Output : 4 5 1) Find average of elements. 2
5 min read