数组-二分查找

1. 一个简单的例题

力扣题目链接

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

示例 1:

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

示例二：

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

提示：

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

---

思路

这道题给的数组是有序的，并且元素不重复，第一个想到的肯定是直接一个for循环搞定，从头开始遍历。

但这道题都强调了元素有序且不重复，那么我们就可以考虑考虑二分法了。

二分查找涉及很多边界条件，尤其是对区间的定义。比如是while(left < right) 还是 while(left <= right), 到底是right = middle呢，还是要right = middle - 1呢？确定这个区间很容易搞混乱。在二分查找的过程中，保持不变量，就是在while寻找中每一次边界的处理都要坚持根据区间的定义来操作，这就是循环不变量规则。

二分法，区间一般定义为两种，左闭右闭[left, rigth] ,或者 左闭右开[left, rigth).

1.1 左闭右闭

定义 target 是在一个在左闭右闭的区间里，也就是[left, right]，这样也就决定了两个特点：

• while (left <= right) 要使用 <= ，因为left == right是有意义的，所以使用 <=
• if (nums[middle] > target) right 要赋值为 middle - 1，因为当前这个nums[middle]一定不是target，那么接下来要查找的左区间结束下标位置就是 middle - 1

例如在数组：1,2,3,4,7,9,10中查找元素2，如图所示：

public int search(int[] nums, int target) {
    if (nums == null || nums.length <= 0) return -1;
    int l = 0, h = nums.length - 1; // 定义target在左闭右闭的区间里，[l, h]

while (l <= h) { //当l == h时，区间[l,h]依然有效，所以是 <=
        int middle = l + (h - l) / 2; //防止溢出，等同于(l + h) / 2
        if (nums[middle] < target) {
            l = middle + 1; //target在右区间，所以[middle + 1, h]
        } else if (nums[middle] > target) {
            h = middle - 1; //target 在左区间，所以[l, middle - 1]
        } else {
            return middle; //找到了，返回
        }
    }

    return -1;
}

1.2 左闭右开

定义 target 是在一个在左闭右开的区间里，也就是[left, right) ，那么二分法的边界处理方式则截然不同。有以下两点：

• while (left < right)，这里使用 < ,因为left == right在区间[left, right)是没有意义的
• if (nums[middle] > target) right 更新为 middle，因为当前nums[middle]不等于target，去左区间继续寻找，而寻找区间是左闭右开区间，所以right更新为middle，即：下一个查询区间不会去比较nums[middle]

在数组：1,2,3,4,7,9,10中查找元素2，如图所示：（注意和方法一的区别）

public int search(int[] nums, int target) {
    if (nums == null || nums.length <= 0) return -1;
    int l = 0, h = nums.length; // 定义target在左闭右开的区间里，即：[l, h)

    if (nums[l] > target || nums[h - 1] < target) return -1; //减少二分查找次数

    while (l < h) { //因为l == h的时候，在[l, h)是无效的空间，所以使用 <
        int middle = l + ((h - l) >> 1);
        if (nums[middle] < target) {
            l = middle + 1; // target 在右区间，在[middle, h)中
        } else if (nums[middle] > target) {
            h = middle; // target 在左区间，在[l, middle)中
        } else {
            return middle;
        }
    }

    return -1;
}

2 . 搜索插入位置

力扣题目链接

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

你可以假设数组中无重复元素。

示例 1:

• 输入: [1,3,5,6], 5
• 输出: 2

示例 2:

• 输入: [1,3,5,6], 2
• 输出: 1

示例 3:

• 输入: [1,3,5,6], 7
• 输出: 4

示例 4:

• 输入: [1,3,5,6], 0
• 输出: 0

---

思路

这道题要在数组中插入目标值，无非是四种情况

• 目标值在数组所有元素之前
• 目标值等于数组中某一个元素
• 目标值插入数组中的位置
• 目标值在数组所有元素之后

2.1 暴力解法

public int searchInsert(int[] nums, int target) {
    int pre = 0; //保存当前节点的前一个节点

    if (target < nums[0]) return 0;

    for (int i = 0; i < nums.length; i++) {
        if (nums[i] == target) return i; //第2中情况，找到了，返回
        else if (nums[pre] < target && nums[i] > target) return i; //如果当前节点比target大并且前一个节点比target小，说明当前节点就是要插入的位置，满足情况1和情况3
        else pre = i; //保存前一个节点
    }

    // 目标值在数组所有元素之后的情况
    return nums.length;
}

2.2 左闭右闭区间

本题也是有序且无序，就可以使用二分法来查找。

定义 target 是在一个在左闭右闭的区间里，也就是[left, right] （这个很重要）。

public int searchInsert(int[] nums, int target) {
    int l = 0, h = nums.length - 1;

    while (l <= h) { //定义target在左闭右闭的区间里，[l, h]
        int middle = l + ((h - l) >> 1);
        if (nums[middle] > target) h = middle - 1;
        else if (nums[middle] < target) l = middle + 1;
        else return middle;
    }

    // 分别处理如下四种情况
    // 目标值在数组所有元素之前  [0, -1]
    // 目标值等于数组中某一个元素  return middle;
    // 目标值插入数组中的位置 [left, right]，return  right + 1
    // 目标值在数组所有元素之后的情况 [left, right]， return right + 1
    return h + 1; //目标值在数组最后

2.3 左闭右开区间

定义 target 是在一个在左闭右开的区间里，也就是[left, right) 。

public int searchInsert(int[] nums, int target) {
    int l = 0, h = nums.length;

    while (l < h) { //定义target在左闭右闭的区间里，[l, h)
        int middle = l + ((h - l) >> 1);
        if (nums[middle] > target) h = middle;
        else if (nums[middle] < target) l = middle + 1;
        else return middle;
    }

    // 分别处理如下四种情况
    // 目标值在数组所有元素之前  [0, 0)
    // 目标值等于数组中某一个元素  return middle;
    // 目标值插入数组中的位置 [left, right)，return  right
    // 目标值在数组所有元素之后的情况 [left, right)， return right
    return h; //目标值在数组最后
}

3. 总结

• 对于左闭右闭区间, while (left <= rigth) rigth = middle - 1
• 对于左闭右开区间, while (left < rigth) rigth = middle

以上内容学自代码随想录