LeetCode练习——704.二分查找（C++）

题目链接

二分查找

题目描述

给定一个$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。

提示：

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

思路

二分法思想

二分法是一种在有序数组中查找特定元素的搜索算法，要实现二分法，前提条件是用于搜索的数组必须是有序的而且数组中没有重复元素。
其核心思想是：每次查找缩小一般的查找范围，直到找到目标值或搜索完数组。
步骤：（默认数组是递增的）

• 首先选择数组中间的数字和要查找的目标值进行比较；
• 如果相等，说明找到了我们要找的目标值，直接返回该元素的下标；
• 如果不相等：
  • 如果中间的数字大于目标值，则将搜索范围缩小到中间数字左边的部分，右边部分全部排除。
  • 如果中间的数字小于目标值，则将搜索范围缩小到中间数字右边的部分，左边部分全部排除。
• 对剩余部分重复以上操作，直到找到目标值或者搜索玩数组，方可结束。

二分法区间一般右两种写法，左闭右闭$[left,right]$和左闭右开$[left,right)$。

第一种写法（左闭右闭）即$[left,right]$

• 由于$left==right$是有意义的，所以此时在$while$循环中要使用$<=$，即要用$while(left<=right)$。
• $if(nums[middle]<target)$，$left$要赋值$middle+1$，因为$nums[middle]$一定不是要找的目标值$target$，而且$target$的值一定比$nums[middle]$大。
• 同理$if(nums[middle]>target)$，$right$要赋值$middle-1$，因为$nums[middle]$一定不是要找的目标值$target$，而且$target$的值一定比$nums[middle]$小。

第二种写法（左闭右开）即$[left,right)$

• 由于$left==right$是没有意义的，所以此时在$while$循环中要使用$<$，即要用$while(left<=right)$。
• $if(nums[middle]<target)$，$left$要赋值$middle+1$，因为$nums[middle]$一定不是要找的目标值$target$，而且$target$的值一定比$nums[middle]$大。
• 同理$if(nums[middle]>target)$，$right$要赋值$middle$，因为$nums[middle]$一定不是要找的目标值$target$，而且$target$的值一定比$nums[middle]$小，而且下一个搜索区间不会比较$nums[middle]$，所以$right$需要赋值$middle$。

题解代码

第一种写法

class Solution {
public:
    int search(vector<int>& nums, int target) {
    	// 定义左边界
        int left = 0;
        // 定义有边界
        int right = nums.size() - 1;
		
        while (left <= right) {
            int middle = (left + right) / 2;
            // 中间的数比target小，left赋值middle + 1
            if (nums[middle] < target) {
                left = middle + 1;
            }
            // 中间的数比target大，left赋值middle - 1
            else if (nums[middle] > target) {
                right = middle - 1;
            }
            // 找到目标值，返回该元素在数组中的下标
            else {
                return middle;
            }
        }
        // 目标值不再数组中，返回-1
        return -1;
    }
};

• 时间复杂度为：O(n)
• 空间复杂度为：O(1)

第二种写法

class Solution {
public:
    int search(vector<int>& nums, int target) {
    	// 定义左边界
        int left = 0;
        // 定义有边界
        int right = nums.size();

        while (left < right) {
            int middle = (left + right) / 2;
            // 中间的数比target小，left赋值middle + 1
            if (nums[middle] < target) {
                left = middle + 1;
            }
            // 中间的数比target大，right赋值middle
            else if (nums[middle] > target) {
                right = middle;
            }
            // 找到目标值，返回该元素在数组中的下标
            else {
                return middle;
            }
        }
        // 目标值不再数组中，返回-1
        return -1;
    }
};

• 时间复杂度为：O(n)
• 空间复杂度为：O(1)

总结

二分法是很基础而且很重要的一个算法，在使用时要特别注意区间的选择，以及不同区间循环条件的使用和循环变量的更新。

本文相关信息：
参考视频：哔哩哔哩—代码随想录