Find First and Last Position of Element in Sorted Array in Python

Suppose we have an array of integers A. This is sorted in ascending order, we have to find the starting and ending position of a given target value. When the target is not found in the array, return [-1, -1]. So if the array is like [2,2,2,3,4,4,4,4,5,5,6], and target is 4, then the output will be [4,7]

To solve this, we will follow these steps −

• Initially res := [-1,-1], set low := 0, high := length of array A
• while low < high
  • mid := low + (high – low)/2
  • if A[mid] is target, then
    • high := mid, res[0] := mid and res[1] := mid
  • otherwise when A[mid] < target, then low := mid + 1, else high := mid
• if res[0] = -1, then return res
• low := res[0] + 1, high := length of nums
• while low < high
  • mid := low + (high – low)/2
  • if A[mid] is target, then
    • low := mid + 1, res[1] := mid
  • otherwise when A[mid] < target, then low := mid + 1, else high := mid
• return res

Example(Python)

Let us see the following implementation to get better understanding −

 Live Demo

class Solution(object):
   def searchRange(self, nums, target):
      res = [-1,-1]
      low = 0
      high = len(nums)
      while low<high:
         mid = int(low + (high-low)//2)
         if nums[mid] == target:
            high = mid
            res[0]=mid
            res[1]=mid
         elif nums[mid]<target:
            low = mid+1
         else:
            high = mid
      if res[0] == -1:
         return res
      low = res[0]+1
      high = len(nums)
      while low<high:
         mid = int(low + (high-low)//2)
         if nums[mid] == target:
            low = mid+1
            res[1] = mid
         elif nums[mid] < target:
            low = mid + 1
         else:
            high = mid
      return res
ob1 = Solution()
print(ob1.searchRange([2,2,2,3,3,4,4,4,4,5,5,6], 4))

Input

[2,2,2,3,4,4,4,4,5,5,6]
4

Output

[5, 8]

Arnab Chakraborty

Published on 30-Jan-2020 10:50:43

• Related Questions & Answers
• First element and last element in a JavaScript array?
• Find position of an element in a sorted array of infinite numbers in C++
• Finding first unique element in sorted array in JavaScript
• Find missing element in a sorted array of consecutive numbers in Python
• Maximum difference between first and last indexes of an element in array in C
• Finding the first unique element in a sorted array in JavaScript
• Find missing element in a sorted array of consecutive numbers in C++
• Find element position in given monotonic sequence in Python
• Delete first and last element from a LinkedList in Java
• Get first and last elements of a list in Python
• MongoDB query to find property of first element of array?
• Find the element that appears once in sorted array - JavaScript
• Find last element after deleting every second element in array of n integers in C++
• Find index of an extra element present in one sorted array in C++
• Maximum element in a sorted and rotated array in C++