Skip to main content

Find First and Last Position of Element in Sorted Array with Explanation | Java

Find First and Last Index Occurrence of given target from Java Array using Binary Search

Find First and Last Index Occurrence of given target from Java Array using Binary Search

Problem Description :

Given an array of integers nums sorted in non-decreasing order, find the starting and ending position of a given target value.

If target is not found in the array, return [-1, -1].

Example 1 :

Input: nums = [5, 7, 7, 8, 8, 8, 10], target = 8
Output: [3, 5]

Example 2 :

Input: nums = [5, 7, 7, 8, 8, 8, 10], target = 6
Output: [-1, -1]

We can solve this problem using brute force approach but it will take more time than binary search. So lets see binary search approach.

Solution 1 : Finding First and Last Position of Element in Array.


import
 java.util.Scanner;

public class FindFirstAndLastPosition {

    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);

        System.out.println("Enter array size : ");
        int size = sc.nextInt();

        int[] array = new int[size];

        // Taking user inputs
        System.out.println("Enter array elements :");
        for (int i = 0; i < array.length; i++) {
            array[i] = sc.nextInt();
        }

        System.out.println("Enter target : ");
        int target = sc.nextInt();

        int[] result = new int[2];
       
        result[0] = searchFirstElement(array, target);
        result[1] = searchLastElement(array, target);
       
        System.out.println("["+result[0]+", "+result[1]+"]");

    }

    private static int searchFirstElement(int[] nums, int target) {
       
        int index = -1;
       
        if (nums.length == 0) {
            return index;
        }

        int start = 0;
        int end = nums.length-1;

        while (start <= end) {
            int mid = (end + start) / 2;
           
            if (nums[mid] >= target) {
                end = mid - 1;
            } else {
                start = mid + 1;
            }
           
            if (nums[mid] == target) {
                index = mid;
            }
        }

        return index;
    }

    private static int searchLastElement(int[] nums, int target) {

        int index = -1;
       
        if (nums.length == 0) {
            return index;
        }

        int start = 0;
        int end = nums.length-1;

        while (start <= end) {
            int mid = (end + start) / 2;
           
            if (nums[mid] <= target) {
                start = mid + 1;
            } else {
                end = mid - 1;
            }
           
            if (nums[mid] == target) {
                index = mid;
            }
        }
        return index;
    }
}

Output :

Enter array size :
6
Enter array elements :
1 2 3 8 8 10
Enter target :
8
[3 , 4]

------------------------------

Enter array size :
10
Enter array elements :
1 2 3 5 6 8 10 11 18 20
Enter target :
4
[-1, -1]

Solution Explanation :

  • Here we are using two methods, searchFirstElement for finding first and searchLastElement for finding last elements in given array. In both method, Binary search is used.
  • In searchFirstElement method,
    • We have to search till first element until current array element < target for finding first target.
  • In searchLastElement method,
    • We have to search till last element until current array element > target for finding last given target.
  • At last return index from both methods.

Comments

Popular posts from this blog

Flipping the Matrix HackerRank Solution in Java with Explanation

Java Solution for Flipping the Matrix | Find Highest Sum of Upper-Left Quadrant of Matrix Problem Description : Sean invented a game involving a 2n * 2n matrix where each cell of the matrix contains an integer. He can reverse any of its rows or columns any number of times. The goal of the game is to maximize the sum of the elements in the n *n submatrix located in the upper-left quadrant of the matrix. Given the initial configurations for q matrices, help Sean reverse the rows and columns of each matrix in the best possible way so that the sum of the elements in the matrix's upper-left quadrant is maximal.  Input : matrix = [[1, 2], [3, 4]] Output : 4 Input : matrix = [[112, 42, 83, 119], [56, 125, 56, 49], [15, 78, 101, 43], [62, 98, 114, 108]] Output : 119 + 114 + 56 + 125 = 414 Full Problem Description : Flipping the Matrix Problem Description   Here we can find solution using following pattern, So simply we have to find Max of same number of box like (1,1,1,1). And ...

How to Implement One to Many and Many to One Mapping in Spring Boot using JPA

Spring Boot CRUD example using One-to-Many and Many to One mapping | With Thymeleaf User Interface In this tutorial, we will learn how to use @OneToMany and @ManyToOne annotation using JPA (Java Persistent API) in Spring Boot. We also attach Thymeleaf for User Interface. In past tutorial, we already created Spring Boot CRUD with Rest API, JPA and MySql. Please refer that one first, we will continue from there. Spring Boot application with Thymeleaf, Rest API, JPA and MySql Database    For applying One to Many relationship, we need another POJO class. In past we already created Book class, now we will create new class Author . As we know Author have multiple Books, so we can easily apply One to Many operation. Lets create POJO class for Author and apply @OneToMany on Book .  Define List of Book and apply @OneToMany annotation on field. We are using mappedBy property, so Author table does not create new column.  We already learn about mappedBy property in One-to-One a...