Equilibrium index of an array
Problem Description
You are given an array A of integers of size N.
Your task is to find the equilibrium index of the given array
The equilibrium index of an array is an index such that the sum of elements at lower indexes is equal to the sum of elements at higher indexes.
If there are no elements that are at lower indexes or at higher indexes, then the corresponding sum of elements is considered as 0.
Notes : -
- Array indexing starts from 0.
- If there is no equilibrium index then return -1.
- If there are more than one equilibrium indexes then return the minimum index.
Problem Constraints
1 <= N <= 10^5
-105 <= A[i] <= 10^5
Input Format
First arugment is an array A.
Output Format
Return the equilibrium index of the given array. If no such index is found then return -1.
Example Input
Input 1:
A = [-7, 1, 5, 2, -4, 3, 0]
Input 2:
A = [1, 2, 3]
Example Output
Output 1:
3
Output 2:
-1
Example Explanation
Explanation 1:
i Sum of elements at lower indexes Sum of elements at higher indexes
0 0 7
1 -7 6
2 -6 1
3 -1 -1
4 1 3
5 -3 0
6 0 0
3 is an equilibrium index, because:
A[0] + A[1] + A[2] = A[4] + A[5] + A[6]
Explanation 2:
i Sum of elements at lower indexes Sum of elements at higher indexes
0 0 5
1 1 3
2 3 0
Thus, there is no such index.
Output
Java
public int equilibriumIndex(int[] A) {
int n = A.length;
long totalSum = 0;
long leftSum = 0;
for (int i = 0; i < n; i++) {
totalSum += A[i];
}
for (int i = 0; i < n; i++) {
totalSum -= A[i];
if (leftSum == totalSum) {
return i;
}
leftSum += A[i];
}
return -1;
}
Python
def equilibrium_index(A):
n = len(A)
total_sum = sum(A)
left_sum = 0
for i in range(n):
total_sum -= A[i]
if left_sum == total_sum:
return i
left_sum += A[i]
return -1
JavaScript
function equilibriumIndex(A) {
const n = A.length;
let totalSum = 0;
let leftSum = 0;
for (let i = 0; i < n; i++) {
totalSum += A[i];
}
for (let i = 0; i < n; i++) {
totalSum -= A[i];
if (leftSum === totalSum) {
return i;
}
leftSum += A[i];
}
return -1;
}