leetcode 34.在排序数组中查找元素的第一个和最后一个位置

在排序数组中查找元素的第一个和最后一个位置

leetcode: 34.在排序数组中查找元素的第一个和最后一个位置

题解

题目中说了是一个非递减的排序数组，从中查找给定的元素位置，很容易想到使用二分查找的思路

一、找下界（最左侧的目标元素）

1.初始情况

low指向数组头部 heigh指向数组尾部
mid指向（low + heigh）/ 2 的元素

2.第一次二分

Arr[mid] >= target
说明下界肯定在mid的左边，因为右边全是大于或等于target的元素
将heigh指向mid左边的一个元素

同时用变量temp记录下与target相等的元素的位置 temp = mid

对应修改mid的指向

3.第二次二分

Arr[mid] < target
说明下界肯定在mid的右侧，因为左侧全是小于target的元素
移动low到mid的右边一个元素

4.第三次二分

Arr[mid] >= target
继续寻找可能出现在左边的下界

同时Arr[mid] = target 让temp = mid

但此时heigh指向mid左边的一个元素后，会使得heigh < low

循环退出，返回最终的temp值 2

二、找上界（最右侧的目标元素）

1.初始化

Arr[mid] <= target
同时Arr[mid] == target temp = mid = 4

2.第一次二分

Arr[mid] <= target
target的上界在mid右侧（左侧全是小于或等于target的元素）
移动low指向mid的右侧一个元素

3.第二次二分

Arr[mid] > target
说明上界在mid的左侧部分
收缩heigh到mid的左侧一个元素

4.第三次二分

Arr[mid] > target