算法 —— 二分查找

目录

二分查找

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

搜索插入位置

x的平方根

山峰数组的峰顶索引

寻找峰值

搜索旋转排序数组中的最⼩值

点名

---

 二分查找模板分为三种：1、朴素的二分模板   2、查找左边界的二分模板  3、查找右边界的二分模板（注意：不是数组有序才使用二分查找，只要存在二段性（一个条件把数组分为两段）都可以使用二分查找）

---

二分查找

 代码如下：

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;
            else if (nums[mid] < target)
                left = mid + 1;
            else
                return mid;
        }
        return -1;
    }
};

---

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

这道题可以引出另外两个重要的二分查找模板： 查找左边界的二分模板   查找右边界的二分模板

 以上是两个模板的内容，判断条件根据题目内容修改，以题目示例1为例，下面给出具体解释为什么这样做可行：

 代码如下：

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        // 处理为空
        if (nums.size() == 0)
            return { -1,-1 };
        // 找左端点
        int left_end_point = -1, right_end_point = -1;
        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;
        }
        // 判断是否有结果
        if(nums[left]==target)
            left_end_point = left;

        // 找右端点   // left可以从左端点开始
        left = 0, right = nums.size() - 1;
        while (left < right)
        {
            int mid = left + (right - left + 1) / 2;
            if (nums[mid] > target)
                right = mid - 1;
            else
                left = mid;
        }
        if(nums[right] == target)
            right_end_point = right;
        if(right_end_point != -1)
            return { left_end_point,right_end_point };
        else
            return { -1,-1 };
    }
};

---

搜索插入位置

根据 二段性，可以把数组分为小于t和大于等于t两部分，目标索引就是在大于等于的左边界上。

注意示例3的边界情况，代码如下：

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0, right = nums.size();
        while (left < right)
        {
            int mid = left + (right - left) / 2;
            if (nums[mid] < target)
                left = mid + 1;
            else
                right = mid;
        }
        // 数组中所有元素小于target
        if (nums[left] < target)
            return left + 1;
        return right;
    }
};

---

x的平方根

本题依旧是一个二分查找的算法思想，left为1，right为x本身，根据二段性，将x分为小于等于sqrt（x）的和大于sqrt（x）的，注意小于1的小数和INT_MAX这两个特殊情况， INT_MAX平方后数据太大，要用long long类型来存储。代码如下：

class Solution {
public:
    int mySqrt(int x) {
        // 处理边界情况
        if (x < 1)
            return 0;
        int left = 1, 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;
    }
};

---

山峰数组的峰顶索引

本题依旧是一道二分查找题，数组被分为递增段和递减端两部分，代码如下：

class Solution {
public:
    int peakIndexInMountainArray(vector<int>& arr) {
        int left = 1, right = arr.size() - 2;
        while (left < right)
        {
            int mid = left + (right - left + 1) / 2;
            if (arr[mid] < arr[mid - 1])
                right = mid - 1;
            else
                left = mid;
        }
        return left;
    }
};

---

寻找峰值

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])
                left = mid + 1;
            else
                right = mid;
        }
        return left;
    }
};

---

搜索旋转排序数组中的最⼩值

class Solution {
public:
    int findMin(vector<int>& nums) {
        int left = 0, right = nums.size() - 1, target = nums[right];
        while (left < right)
        {
            int mid = left + (right - left) / 2;
            if (nums[mid] > target)
                left = mid + 1;
            else
                right = mid;
        }
        return nums[right];
    }
};

---

点名

 本题可以有多种解法：

此题查找的是左边界，直接写代码即可：

class Solution {
public:
    int takeAttendance(vector<int>& records) {
        int left = 0, right = records.size() - 1;
        while (left < right)
        {
            int mid = left + (right - left) / 2;
            if (records[mid] == mid)
                left = mid + 1;
            else
                right = mid;
        }
        // 特殊情况0 1 2 3 缺少4
        return records[left] == left ? left + 1 : left;
    }
};