Find Pivot Index Java Solution - The Coding Shala

Home >> Interview Questions >> Find Pivot Index

Find Pivot Index

Problem:
Given an array of integers nums, write a method that returns the "pivot" index of this array.

We define the pivot index as the index where the sum of the numbers to the left of the index is equal to the sum of the numbers to the right of the index.

If no such index exists, we should return -1. If there are multiple pivot indexes, you should return the left-most pivot index.

Example 1:

Input: 
nums = [1, 7, 3, 6, 5, 6]
Output: 3
Explanation: 
The sum of the numbers to the left of index 3 (nums[3] = 6) is equal to the sum of numbers to the right of index 3.
Also, 3 is the first index where this occurs.

Example 2:

Input: 
nums = [1, 2, 3]
Output: -1
Explanation: 
There is no index that satisfies the conditions in the problem statement.

Find Pivot Index Java Solution

Approach:

At every index, we can check if the total sum to its left and total sum to right of this index equals or not.

First, we will find the total sum of an array then using a loop every time we check if the sum till the current index-1 is equaled to the sum to the right side of this index.

Java Code:: 

class Solution {
    public int pivotIndex(int[] nums) {
        int len = nums.length;
        if(len<3) return -1;
        int right_sum = 0;
        //total sum
        for(int i=0; i<len; i++) right_sum += nums[i];
        
        int left_sum = 0;
        //checking sum
        for(int i=0;i<len;i++){
            right_sum -= nums[i]; //current index does not count
            if(left_sum == right_sum) return i;
            left_sum += nums[i];
        }
        
        return -1;
    }
}


Other Posts You May Like
Please leave a comment below if you like this post or found some errors, it will help me to improve my content.

Comments

Popular Posts from this Blog

Java Program to Convert Binary to Decimal - The Coding Shala

N-th Tribonacci Number Solution - The Coding Shala

Shell Script to find sum, product and average of given numbers - The Coding Shala

Shell Script to Create a Simple Calculator - The Coding Shala

Java Program to Convert Decimal to Binary - The Coding Shala