有序数组和二分查找法

1 篇文章 0 订阅
1 篇文章 0 订阅

前言:

做leetcode第35题搜索插入位置题时,要求用O(lgn)时间度的方法,所以不能用暴力求解法。所以可以尝试一下二分法,但是之前每次做二分法都要分析好久,看网上别人的做法,每次思路差不多,但细节总是差那么一点,也不懂为什么要这么改。通过评论区carl大神的讲解,算是明白些了,借此总结一下。
明白之后可以写一下704题二分查找

二分法

这道题目的前提是数组是有序数组,这也是使用二分查找的基础条件。
以后大家只要看到面试题里给出的数组是有序数组,都可以想一想是否可以使用二分法。

同时题目还强调数组中无重复元素,因为一旦有重复元素,使用二分查找法返回的元素下表可能不是唯一的(建议熟悉二分法之后,尝试leetcode第34题,就是这种有重复元素的情况,我做的时候使用二分法+递归,分左半边和右半边单独考虑的)。

二分查找涉及的很多的边界条件,逻辑比较简单,就是写不好。主要是对二分查找法中边界条件处理不好。例如到底是 while(left < right) 还是 while(left <= right),到底是right = middle呢,还是要right = middle - 1呢?

这里弄不清楚主要是因为对区间的定义没有想清楚,这就是不变量
要在二分查找的过程中,保持不变量,这也就是循环不变量 (感兴趣的同学可以查一查)。

程序以第35题要求来写的

第一种写法:

以这道题目来举例,以下的代码中定义 target 是在一个在左闭右闭的区间里,也就是[left, right] (这个很重要)

这就决定了这个二分法的代码如何去写,大家看如下代码:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0;
        int right = n - 1; // 定义target在左闭右闭的区间里,[left, right] 
        while (left <= right) { // 当left==right,区间[left, right]依然有效
            int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
            if (nums[middle] > target) {
                right = middle - 1; // target 在左区间,所以[left, middle - 1]
            } else if (nums[middle] < target) {
                left = middle + 1; // target 在右区间,所以[middle + 1, right]
            } else { // nums[middle] == target
                return middle;
            }
        }
        // 分别处理如下四种情况
        // 目标值在数组所有元素之前  [0, -1]
        // 目标值等于数组中某一个元素  return middle;
        // 目标值插入数组中的位置 [left, right],return  right + 1
        // 目标值在数组所有元素之后的情况 [left, right], return right + 1
        return right + 1;
    }
};

大家要仔细看注释,要写while(left <= right)和right = middle - 1
因为区间左闭右闭,要包含右边的元素一定可以考虑到,所以middle-1,可以拿例子试一试

第二种写法:

如果说定义 target 是在一个在左闭右开的区间里,也就是[left, right) 。
那么二分法的边界处理方式则截然不同。
不变量是[left, right)的区间,如下代码可以看出是如何在循环中坚持不变量的。

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0;
        int right = n; // 定义target在左闭右开的区间里,[left, right)  target
        while (left < right) { // 因为left == right的时候,在[left, right)是无效的空间
            int middle = left + ((right - left) >> 1);
            if (nums[middle] > target) {
                right = middle; // target 在左区间,在[left, middle)中
            } else if (nums[middle] < target) {
                left = middle + 1; // target 在右区间,在 [middle+1, right)中
            } else { // nums[middle] == target
                return middle; // 数组中找到目标值的情况,直接返回下标
            }
        }
        // 分别处理如下四种情况
        // 目标值在数组所有元素之前 [0,0)
        // 目标值等于数组中某一个元素 return middle
        // 目标值插入数组中的位置 [left, right) ,return right 即可
        // 目标值在数组所有元素之后的情况 [left, right),return right 即可
        return right;
    }
};

大家要仔细看注释,要写while (left < right)和right = middle。因为右边是开区间,所以取不到该元素,right=middle,就不再-1.

两种方法的时间复杂的都是O(logn)


参考链接:
https://mp.weixin.qq.com/s/_DzddsMeQW5JPI6qoC7ARQ

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值