【算法专题】二分查找

2024-03-01 11:22:44 浏览数 (1)

二分查找

1. 二分查找

题目链接 -> Leetcode -704.二分查找

Leetcode -704.二分查找

题目:给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 - 1。

示例 1: 输入: nums = [-1, 0, 3, 5, 9, 12], target = 9 输出 : 4 解释 : 9 出现在 nums 中并且下标为 4

示例 2 : 输入 : nums = [-1, 0, 3, 5, 9, 12], target = 2 输出 : -1 解释 : 2 不存在 nums 中因此返回 - 1

提示: 你可以假设 nums 中的所有元素是不重复的。 n 将在[1, 10000]之间。 nums 的每个元素都将在[-9999, 9999]之间。

思路:

  1. 定义 left , right 指针,分别指向数组的左右区间。
  2. 找到待查找区间的中间点 mid ,找到之后分三种情况讨论:
  • arr[mid] == target 说明正好找到,返回 mid 的值;
  • arr[mid] > target 说明 [mid, right] 这段区间都是大于 target 的,因此舍去右边区间,在左边 [left, mid -1] 的区间继续查找,即让 right = mid - 1 ,然后重复 2 过程;
  • arr[mid] < target 说明 [left, mid] 这段区间的值都是小于 target 的,因此舍去左边区间,在右边 [mid 1, right] 区间继续查找,即让 left = mid 1 ,然后重复 2 过程;
  1. 当 left 与 right 错开时,说明整个区间都没有这个数,返回 -1 。

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    int search(vector<int>& nums, int target)
		    {
		        int left = 0, right = nums.size() - 1;
		        while (left <= right)
		        {
		            int mid = left   (right - left) / 2;  // 防止溢出
		            if (nums[mid] > target) right = mid - 1; // 中间数比 target 大,改变右指针
		            else if (nums[mid] < target) left = mid   1; // 中间数比 target 小,改变左指针
		            else return mid; // 找到返回下标
		        }
		        return -1;
		    }
		};

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

题目链接 -> Leetcode -34.在排序数组中查找元素的第一和最后一个位置

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

题目:给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target,返回[-1, -1]。

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 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]

提示:

  • 0 <= nums.length <= 10^5
  • 10^9 <= nums[i] <= 10^9 nums 是一个非递减数组
  • 10^9 <= target <= 10^9

思路:二分思想,就是根据数据的性质,在某种判断条件下将区间一分为二,然后舍去其中一个区间,然后再另一个区间内查找;为了方便叙述,用 x 表示该元素, resLeft 表示左边界, resRight 表示右边界;

寻找左边界思路: 我们注意到以左边界划分的两个区间的特点:

  • 左边区间 [left, resLeft - 1] 都是小于 x 的;
  • 右边区间(包括左边界) [resLeft, right] 都是大于等于 x 的;

因此,关于 mid 的落点,我们可以分为下面两种情况:

  • 当我们的 mid 落在 [left, resLeft - 1] 区间的时候,也就是 arr[mid] < target 。说明 [left, mid] 都是可以舍去的,此时更新 left 到 mid 1 的位置,继续在 [mid 1, right] 上寻找左边界
  • 当 mid 落在 [resLeft, right] 的区间的时候,也就是 arr[mid] >= target 。说明 [mid 1, right] (因为 mid 可能是最终结果,不能舍去)是可以舍去的,此时更新 right 到 mid 的位置,继续在 [left, mid] 上寻找左边界;

注意:这里找中间元素需要向下取整。因为后续移动左右指针的时候:

  • 左指针: left = mid 1 ,是会向后移动的,因此区间是会缩小的;
  • 右指针: right = mid ,可能会原地踏步(比如:如果向上取整的话,如果剩下 1,2 两个元素, left == 1 , right == 2 , mid == 2 。更新区间之后, left,right,mid 的值没有改变,就会陷入死循环)。因此一定要注意,当 right = mid 的时候,要向下取整。

同理大家可以尝试分析寻找右边界的思路。

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    vector<int> searchRange(vector<int>& nums, int target)
		    {
		        if (nums.size() == 0) return { -1, -1 };
		
		        int left = 0, right = nums.size() - 1;
		        int begin = -1;
		
		        // 寻找左端点
		        while (left < right)
		        {
		            // 当元素个数为偶数个时,取左边的为中间元素,因为要找的是左边界
		            int mid = left   (right - left) / 2;  
		            if (nums[mid] < target) left = mid   1;
		            else right = mid;
		        }
		        if (target == nums[left]) begin = left;
		        else return { -1, -1 };
		
		        left = 0, right = nums.size() - 1;
		        // 寻找右端点
		        while (left < right)
		        {
		            // 当元素个数为偶数个时,取右边的为中间元素,因为要找的是右边界
		            // 或者当更新 right / left 时出现 -1 取mid的时候就要 1
		            int mid = left   (right - left   1) / 2;  
		            if (nums[mid] > target) right = mid - 1;
		            else left = mid;
		        }
		        return { begin, right };
		    }
		};

3. 搜索插入位置

题目链接 -> Leetcode -35.搜索插入位置

Leetcode -35.搜索插入位置

题目:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例 1: 输入: nums = [1, 3, 5, 6], target = 5 输出 : 2

示例 2 : 输入 : nums = [1, 3, 5, 6], target = 2 输出 : 1

示例 3 : 输入 : nums = [1, 3, 5, 6], target = 7 输出 : 4

提示 :

  • 1 <= nums.length <= 10^4
  • 10^4 <= nums[i] <= 10^4 nums 为 无重复元素 的 升序 排列数组
  • 10^4 <= target <= 10^4

思路与上题思路类似,寻找目标值的左边界即可。

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    int searchInsert(vector<int>& nums, int target)
		    {
		        // 如果插入的位置在最后
		        if (nums[nums.size() - 1] < target) return nums.size();
		
		        int left = 0, right = nums.size() - 1;
		
		        // 二分查找找左边界
		        while (left < right)
		        {
		            int mid = left   (right - left) / 2;
		            if (nums[mid] < target) left = mid   1;
		            else right = mid;
		        }
		        return left;
		    }
		};

4. x 的平方根

题目链接 -> Leetcode -69.x 的平方根

Leetcode -69.x 的平方根

题目:给你一个非负整数 x ,计算并返回 x 的 算术平方根 。

由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。

注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x** 0.5 。

示例 1: 输入:x = 4 输出:2

示例 2: 输入:x = 8 输出:2 解释:8 的算术平方根是 2.82842…, 由于返回类型是整数,小数部分将被舍去。

提示: 0 <= x <= 2^31 - 1

思路也和上题类似,寻找右边界即可;

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    int mySqrt(int x)
		    {
		        // long long 防止溢出
		        long long left = 0, right = x;
		
		        // 二分查找寻找右边界
		        while (left < right)
		        {
		            long long mid = left   (right - left   1) / 2;
		            if (mid * mid > x) right = mid - 1;
		            else left = mid;
		        }
		        return left;
		    }
		};

5. 山脉数组的峰顶索引

题目链接 -> Leetcode -825.山脉数组的峰顶索引

Leetcode -825.山脉数组的峰顶索引

题目:符合下列属性的数组 arr 称为 山脉数组 : arr.length >= 3 存在 i(0 < i < arr.length - 1)使得: arr[0] < arr[1] < … arr[i - 1] < arr[i] arr[i] > arr[i 1] > … > arr[arr.length - 1] 给你由整数组成的山脉数组 arr ,返回满足 arr[0] < arr[1] < … arr[i - 1] < arr[i] > arr[i 1] > … > arr[arr.length - 1] 的下标 i 。 你必须设计并实现时间复杂度为 O(log(n)) 的解决方案。

示例 1: 输入:arr = [0, 1, 0] 输出:1

示例 2: 输入:arr = [0, 2, 1, 0] 输出:1

示例 3: 输入:arr = [0, 10, 5, 2] 输出:1

提示: 3 <= arr.length <= 10^5 0 <= arr[i] <= 10^6 题目数据保证 arr 是一个山脉数组

思路:

  1. 分析峰顶位置的数据特点,以及山峰两旁的数据的特点:
  • 峰顶数据特点: arr[i] > arr[i - 1] && arr[i] > arr[i 1] ;
  • 峰顶左边的数据特点: arr[i] > arr[i - 1] && arr[i] < arr[i 1] ,也就是呈现上升趋势;
  • 峰顶右边数据的特点: arr[i] < arr[i - 1] && arr[i] > arr[i 1] ,也就是呈现下降趋势。
  1. 因此,根据 mid 位置的信息,我们可以分为下面三种情况:
  • 如果 mid 位置呈现上升趋势,说明我们接下来要在 [mid 1, right] 区间继续搜索;
  • 如果 mid 位置呈现下降趋势,说明我们接下来要在 [left, mid - 1] 区间搜索;
  • 如果 mid 位置就是山峰,直接返回结果。

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    int peakIndexInMountainArray(vector<int>& arr) 
		    {
		        int left = 0,right = arr.size() - 1;
		        // 符合二段性,就可以用二分查找
		        while(left < right)
		        {
		            int mid = left   (right - left   1)/2;
		            if(arr[mid] > arr[mid - 1]) left = mid;
		            else right = mid - 1;
		        }
		        return left;
		    }
		};

6. 寻找峰值

题目链接 -> Leetcode -162.寻找峰值

Leetcode -162.寻找峰值

题目:峰值元素是指其值严格大于左右相邻值的元素。

给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。

你可以假设 nums[-1] = nums[n] = -∞ 。

你必须实现时间复杂度为 O(log n) 的算法来解决此问题。

示例 1: 输入:nums = [1, 2, 3, 1] 输出:2 解释:3 是峰值元素,你的函数应该返回其索引 2。

示例 2: 输入:nums = [1, 2, 1, 3, 5, 6, 4] 输出:1 或 5 解释:你的函数可以返回索引 1,其峰值元素为 2; 或者返回索引 5, 其峰值元素为 6。

提示:

  • 1 <= nums.length <= 1000
  • 2^31 <= nums[i] <= 2^31 - 1
  • 对于所有有效的 i 都有 nums[i] != nums[i 1]

思路:寻找二段性: 任取⼀个点 i ,与下一个点 i 1 ,会有如下两种情况:

  • arr[i] > arr[i 1] :此时「左侧区域」一定会存在山峰(因为最左侧是负无穷,但此时还没有下降趋势),那么我们可以去左侧去寻找结果;
  • arr[i] < arr[i 1] :此时「右侧区域」一定会存在山峰(因为最右侧是负无穷,但此时还没有下降趋势),那么我们可以去右侧去寻找结果。

当我们找到「二段性」的时候,就可以尝试用「二分查找」算法来解决问题。

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    int findPeakElement(vector<int>& nums) 
		    {
		        int left = 0, right = nums.size() - 1;
		        // 符合二段性,可以用二分查找
		        while(left < right)
		        {
		            int mid = left   (right - left)/2;
		            if(nums[mid] > nums[mid   1]) right = mid;
		            else left = mid   1;
		        }
		        return left;
		    }
		};

7. 寻找旋转排序数组中的最小值

题目链接 -> Leetcode -153.寻找旋转排序数组中的最小值

Leetcode -153.寻找旋转排序数组中的最小值

题目:已知一个长度为 n 的数组,预先按照升序排列,经由 1 到 n 次 旋转 后,得到输入数组。例如,原数组 nums = [0, 1, 2, 4, 5, 6, 7] 在变化后可能得到: 若旋转 4 次,则可以得到[4, 5, 6, 7, 0, 1, 2] 若旋转 7 次,则可以得到[0, 1, 2, 4, 5, 6, 7] 注意,数组[a[0], a[1], a[2], …, a[n - 1]] 旋转一次 的结果为数组[a[n - 1], a[0], a[1], a[2], …, a[n - 2]] 。

给你一个元素值 互不相同 的数组 nums ,它原来是一个升序排列的数组,并按上述情形进行了多次旋转。请你找出并返回数组中的 最小元素 。

你必须设计一个时间复杂度为 O(log n) 的算法解决此问题。

示例 1: 输入:nums = [3, 4, 5, 1, 2] 输出:1 解释:原数组为[1, 2, 3, 4, 5] ,旋转 3 次得到输入数组。

示例 2: 输入:nums = [4, 5, 6, 7, 0, 1, 2] 输出:0 解释:原数组为[0, 1, 2, 4, 5, 6, 7] ,旋转 4 次得到输入数组。

示例 3: 输入:nums = [11, 13, 15, 17] 输出:11 解释:原数组为[11, 13, 15, 17] ,旋转 4 次得到输入数组。

提示:

  • n == nums.length
  • 1 <= n <= 5000
  • 5000 <= nums[i] <= 5000 nums 中的所有整数 互不相同 nums 原来是一个升序排序的数组,并进行了 1 至 n 次旋转

思路:题目中的数组规则如下图所示:

在这里插入图片描述在这里插入图片描述

其中 C 点就是我们要求的点。 二分的本质:找到一个判断标准,使得查找区间能够一分为二。 通过图像我们可以发现, [A,B] 区间内的点都是严格大于 D 点的值的, C 点的值是严格小于 D 点的值的。但是当 [C,D] 区间只有一个元素的时候, C 点的值是可能等于 D 点的值的。

因此,初始化左右两个指针 left , right : 然后根据 mid 的落点,我们可以这样划分下一次查询的区间:

  • 当 mid 在 [A,B] 区间的时候,也就是 mid 位置的值严格大于 D 点的值,下一次查询区间在 [mid 1,right] 上;
  • 当 mid 在 [C,D] 区间的时候,也就是 mid 位置的值严格小于等于 D 点的值,下次查询区间在 [left,mid] 上。

当区间长度变成 1 的时候,就是我们要找的结果。

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    int findMin(vector<int>& nums) 
		    {
		        int left = 0, right = nums.size() - 1;
		        while(left < right)
		        {
		            // 以最右边的元素为目标进行比较
		            int mid = left   (right - left )/2;
		            if(nums[mid] > nums[nums.size() - 1]) left = mid   1;
		            else right = mid;
		        }
		        return nums[left];
		    }
		};

8. 点名

题目链接 -> Leetcode -LCR 173.点名

Leetcode -LCR 173.点名

题目:某班级 n 位同学的学号为 0 ~n - 1。点名结果记录于升序数组 records。假定仅有一位同学缺席,请返回他的学号。

示例 1: 输入: records = [0, 1, 2, 3, 5] 输出 : 4

示例 2 : 输入 : records = [0, 1, 2, 3, 4, 5, 6, 8] 输出 : 7

提示: 1 <= records.length <= 10000

思路是以数组下标为目标对象进行比较,寻找左边界即可;

代码如下:

代码语言:javascript复制
		class Solution {
		public:
		    int missingNumber(vector<int>& nums) 
		    {
		        int left = 0, right = nums.size() - 1;
		        while(left < right)
		        {
		            // 以数组下标为目标对象进行比较
		            int mid = left   (right - left)/2;
		            if(nums[mid] > mid) right = mid;
		            else left = mid   1;
		        }
		        return left == nums[left]? left   1 : left;
		    }
		};

0 人点赞