algorithm-base/animation-simulation/二分查找及其变种/leetcode34查找第一个位置和最后一个位置.md

211 lines
8.2 KiB
Java
Raw Normal View History

2021-07-23 15:44:19 +00:00
> 如果阅读时发现错误或者动画不可以显示的问题可以添加我微信好友 **[tan45du_one](https://raw.githubusercontent.com/tan45du/tan45du.github.io/master/个人微信.15egrcgqd94w.jpg)** ,备注 github + 题目 + 问题 向我反馈
2021-03-20 09:16:07 +00:00
>
> 感谢支持该仓库会一直维护希望对各位有一丢丢帮助
>
2021-07-23 15:44:19 +00:00
> 另外希望手机阅读的同学可以来我的 <u>[**公众号袁厨的算法小屋**](https://raw.githubusercontent.com/tan45du/test/master/微信图片_20210320152235.2pthdebvh1c0.png)</u> 两个平台同步,想要和题友一起刷题,互相监督的同学,可以在我的小屋点击<u>[**刷题小队**](https://raw.githubusercontent.com/tan45du/test/master/微信图片_20210320152235.2pthdebvh1c0.png)</u>进入。
2021-03-20 09:16:07 +00:00
2021-03-18 05:33:53 +00:00
## 查找元素第一个位置和最后一个位置
上面我们说了如何使用二分查找在数组或区间里查出特定值的索引位置但是我们刚才数组里面都没有重复值查到返回即可那么我们思考一下下面这种情况
![二分查找变种一](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/二分查找变种一.3axfe8rq0ei0.png)
此时我们数组里含有多个 5 我们查询是否含有 5 可以很容易查到但是我们想获取第一个 5 最后一个 5 的位置应该怎么实现呢我们可以使用遍历当查询到第一个 5 我们设立一个指针进行定位然后到达最后一个 5 时返回这样我们就能求的第一个和最后一个五了因为我们这个文章的主题就是二分查找我们可不可以用二分查找来实现呢当然是可以的
2021-03-20 09:16:07 +00:00
#### [34. 在排序数组中查找元素的第一个和最后一个位置](https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/)
2021-03-18 05:33:53 +00:00
#### 题目描述
> 给定一个按照升序排列的整数数组 nums和一个目标值 target找出给定目标值在数组中的开始位置和结束位置
>
> 如果数组中不存在目标值 target返回 [-1, -1]
示例 1
> 输入nums = [5,7,7,8,8,10], target = 8
> 输出[3,4]
示例 2
> 输入nums = [5,7,7,8,8,10], target = 6
> 输出[-1,-1]
示例 3
> 输入nums = [], target = 0
> 输出[-1,-1]
#### 题目解析
这个题目很容易理解我们在上面说了如何使用遍历解决该题但是这个题目的目的就是让我们使用二分查找我们来逐个分析先找出目标元素的下边界那么我们如何找到目标元素的下边界呢
我们来重点分析一下刚才二分查找中的这段代码
```java
if (nums[mid] == target) {
return mid;
}else if (nums[mid] < target) {
//这里需要注意,移动左指针
left = mid + 1;
}else if (nums[mid] > target) {
//这里需要注意,移动右指针
right = mid - 1;
2021-07-23 15:44:19 +00:00
}
2021-03-18 05:33:53 +00:00
```
2021-07-23 15:44:19 +00:00
我们只需在这段代码中修改即可我们再来剖析一下这块代码nums[mid] == target 时则返回nums[mid] < target 时则移动左指针在右区间进行查找 nums[mid] > target 时则移动右指针在左区间内进行查找
2021-03-18 05:33:53 +00:00
那么我们思考一下如果此时我们的 nums[mid] = target ,但是我们不能确定 mid 是否为该目标数的左边界所以此时我们不可以返回下标例如下面这种情况![二分查找下边界](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/二分查找下边界.m9m083jempc.png)
此时 mid = 4 nums[mid] = 5,但是此时的 mid 指向的并不是第一个 5所以我们需要继续查找 因为我们要找
2021-07-23 15:44:19 +00:00
的是数的下边界所以我们需要在 mid 的值的左区间继续寻找 5 那我们应该怎么做呢我们只需在
2021-03-18 05:33:53 +00:00
2021-07-23 15:44:19 +00:00
target <= nums[mid] right = mid - 1 即可这样我们就可以继续在 mid 的左区间继续找 5 是不是听着有点绕我们通过下面这组图进行描述
2021-03-18 05:33:53 +00:00
![左边界1](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/左边界1.5o2ihokjfg80.png)
![左边界2](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/左边界2.5wazlfm298s0.png)
2021-07-23 15:44:19 +00:00
其实原理很简单就是我们将小于和等于合并在一起处理 target <= nums[mid] 我们都移动右指针也就是 right = mid -1还有一个需要注意的就是我们计算下边界时最后的返回值为 left 当上图结束循环时left = 3right = 2返回 left 刚好时我们的下边界我们来看一下求下边界的具体执行过程
2021-03-18 05:33:53 +00:00
**动图解析**
![二分查找下边界](https://cdn.jsdelivr.net/gh/tan45du/photobed@master/photo/二分查找下边界.u1cidx99yio.gif)
```java
int lowerBound(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
//这里需要注意计算mid
int mid = left + ((right - left) >> 1);
if (target <= nums[mid]) {
//当目标值小于等于nums[mid]时,继续在左区间检索,找到第一个数
right = mid - 1;
}else if (target > nums[mid]) {
//目标值大于nums[mid]时,则在右区间继续检索,找到第一个等于目标值的数
left = mid + 1;
}
}
return left;
}
```
计算上边界时算是和计算上边界时条件相反
2021-07-23 15:44:19 +00:00
计算下边界时 target <= nums[mid] right = mid -1target > nums[mid] left = mid + 1
2021-03-18 05:33:53 +00:00
2021-07-23 15:44:19 +00:00
计算上边界时 target < nums[mid] right = mid -1; target >= nums[mid] left = mid + 1;刚好和计算下边界时条件相反返回 right
2021-03-18 05:33:53 +00:00
**计算上边界代码**
```java
int upperBound(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
//求mid
int mid = left + ((right - left) >> 1);
//移动左指针情况
if (target >= nums[mid]) {
2021-07-23 15:44:19 +00:00
left = mid + 1;
2021-03-18 05:33:53 +00:00
//移动右指针情况
}else if (target < nums[mid]) {
right = mid - 1;
}
2021-07-23 15:44:19 +00:00
2021-03-18 05:33:53 +00:00
}
return left;
}
```
#### **题目完整代码**
```java
class Solution {
public int[] searchRange (int[] nums, int target) {
int upper = upperBound(nums,target);
2021-07-23 15:44:19 +00:00
int low = lowerBound(nums,target);
2021-03-18 05:33:53 +00:00
//不存在情况
if (upper < low) {
return new int[]{-1,-1};
}
return new int[]{low,upper};
}
//计算下边界
int lowerBound(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
//这里需要注意计算mid
int mid = left + ((right - left) >> 1);
if (target <= nums[mid]) {
//当目标值小于等于nums[mid]时,继续在左区间检索,找到第一个数
right = mid - 1;
}else if (target > nums[mid]) {
//目标值大于nums[mid]时,则在右区间继续检索,找到第一个等于目标值的数
left = mid + 1;
}
}
return left;
}
//计算上边界
int upperBound(int[] nums, int target) {
int left = 0, right = nums.length - 1;
2021-07-23 15:44:19 +00:00
while (left <= right) {
2021-03-18 05:33:53 +00:00
int mid = left + ((right - left) >> 1);
if (target >= nums[mid]) {
2021-07-23 15:44:19 +00:00
left = mid + 1;
2021-03-18 05:33:53 +00:00
}else if (target < nums[mid]) {
right = mid - 1;
2021-07-23 15:44:19 +00:00
}
2021-03-18 05:33:53 +00:00
}
return right;
}
}
```
2021-07-27 18:26:32 +00:00
Go Code:
```go
func searchRange(nums []int, target int) []int {
upper := upperBound(nums, target)
lower := lowerBound(nums, target)
if (upper < lower) {
return []int{-1, -1}
}
return []int{lower, upper}
}
// upperBound 计算上边界
func upperBound(nums []int, target int) int {
l, r := 0, len(nums) - 1
for l <= r {
m := l + (r - l) / 2
if target >= nums[m] {
l = m + 1
} else if target < nums[m] {
r = m - 1
}
}
return r
}
// lowerBound 计算下边界
func lowerBound(nums []int, target int) int {
l, r := 0, len(nums) - 1
for l <= r {
m := l + (r - l) / 2
if target <= nums[m] {
r = m - 1
} else if target > nums[m] {
l = m + 1
}
}
return l
}
```