【算法】二分查找

一、算法描述

需求:在有序数组 A 内,查找值 target;如果找到返回索引,如果找不到返回 -1

1.描述

给定一个含 n 个元素的有序数组 A,满足 A0 ≤ A1 ≤ A2 ≤···≤ An-1,一个待查值 target。

2.解题思路

因为数组是有序的,所以用区间减半来提高查找效率,左区间起点i,右区间起点j,一开始是全部元素,每次拿中间的元素来和目标元素比较,如果相等则返回;
如果目标元素小于中间元素,则说明目标元素在左区间,直接把区间范围缩减到左区间。
如果目标元素大于中间元素,则说明目标元素在右区间,直接把区间范围缩减到右区间。
一直重复,直到找到目标元素。

3.解题步骤

  • 1.设置 i= 0, j=n-1
  • 2.如果 i > j,结束查找,没找到
  • 3.设置 m = floor((i+j)/2),m 为中间索引,floor 是向下取整( ≤(i+j)/2的最小整数)
  • 4.如果 target < Am , 设置 j = m - 1,跳到第2步
  • 5.如果 target > Am , 设置 i = m + 1,跳到第2步
  • 6.如果 target = Am,找到了,结束查找。

二、使用左闭右闭

    /**
     * 二分查找基础版
     * @param arr
     * @param target
     * @return
     */
    public static int binarySearchBasic(int[] arr,int target) {
        int i = 0;
        int j = arr.length - 1;

        while (i <= j) {
            int mid = (i + j) / 2;
            if (target < arr[mid]) {
                // 去到左区间查找
                j = mid - 1;
            } else if (target > arr[mid]) {
                // 去到右区间查找
                i = mid + 1;
            } else {
                return mid;
            }
        }
        return -1;
    }

判断的时候使用的是i <= j,如果没有等于的话,那么参与比较的只有i和j中间的元素,i和j本身就没有参与比较,如果目标就是i和j,则会返回-1。

三、越界问题

当数组的数量达到了整数最大值个数的时候,使用(i+j)/2 运算,当要查找的目标在右区间时,运算的过程中i+j超过了整数的最大值,那么结果会变成负数。
在Java中,数字的表示都是带符号位的,虽然i+j已经超过了整数最大的表示范围,但(i+j)/2并没有超过,所以可以把除以2替换为右移运算,虽然i+j是负数,但是实际他对应的二进制数是没有问题的,只是在Java中,是带符号位的,所以就表现了负数,在使用右移替换了除法之后,得到的数就没有超过最大值了,就可以准确的转换为正确的数字了。
优化代码:

    /**
     * 二分查找基础版
     * @param arr
     * @param target
     * @return
     */
    public static int binarySearchBasic(int[] arr,int target) {
        int i = 0;
        int j = arr.length - 1;

        while (i <= j) {
            int mid = (i + j) >>> 1;
            if (target < arr[mid]) {
                // 去到左区间查找
                j = mid - 1;
            } else if (target > arr[mid]) {
                // 去到右区间查找
                i = mid + 1;
            } else {
                return mid;
            }
        }
        return -1;
    }

四、使用左闭右开

让j只作为边界,而不作为查找目标。

    /**
     * 二分查找基础版 左闭右开
     * @param arr
     * @param target
     * @return
     */
    public static int binarySearch(int[] arr,int target) {
        int i = 0;
        int j = arr.length; // 第一处不同

        while (i < j) {    // 第二处不同
            int mid = (i + j) >>> 1;
            if (target < arr[mid]) {
                // 去到左区间查找
                j = mid;   // 第三处不同
            } else if (target > arr[mid]) {
                // 去到右区间查找
                i = mid + 1;
            } else {
                return mid;
            }
        }
        return -1;
    }

注意:这里的判断,一定要使用<,不能是<=,因为j不作为比较的元素了,只作为边界,否则会在没有元素的情况下进入死循环。

五、平衡二分查找

在上面的写法中:
如果要查找的元素在最左边,那么判断中,只需要执行if判断,不需要执行else if判断,一共执行的次数是n次;
如果要查找的元素在最右边,那么判断中,if判断执行了,但是没满足,else if判断也执行,满足,所以一共执行的次数是2n次。
所以查找并不平衡。

优化思路:把多执行的else if判断抽出去,让每次操作都只执行一次判断,那么比较次数就不存在不平衡的情况了。

/**
     * 平衡二分查找
     * @param arr
     * @param target
     * @return
     */
    public static int balanceBinarySearch(int[] arr, int target) {
        int i = 0;
        int j = arr.length;

        while (j - i > 1) {
            //当i和j中间还有元素的时候就继续
            int mid = (i + j) >>> 1;
            if (target < arr[mid]) {
                j = mid;
            } else {
                i = mid;
            }
        }
        if (arr[i] == target) {
            return i;
        } else {
            return -1;
        }
    }

1.左闭右开的区间,i 指向的可能是目标,而 j 指向的不是目标
2.不在循环内找出目标对象,而是等范围内只剩i时,退出循环,在循环外比较 a[i] 与 target
3.循环内的平均比较次数减少了
4.时间复杂度 θ(log(n))

缺点:无论什么情况,都要遍历完直到只剩下最终的元素,才会退出循环拿到结果。时间复杂度最好和最快都是θ(log(n))

六、使用递归实现

    /**
     * 二分查找
     * @param nums
     * @param target
     * @return
     */
    public int binarySearch(int[] nums, int target) {
        return binarySearchRecursion(nums,target,0,nums.length-1);
    }

    /**
     * 递归查找
     * @param arr
     * @param target
     * @param startIndex
     * @param endIndex
     * @return
     */
    public static int binarySearchRecursion(int[] arr, int target, int startIndex, int endIndex) {
        int mid = (startIndex + endIndex) >>> 1;
        if (startIndex > endIndex) {
            return -1;
        }
        if (target == arr[mid]) {
            return mid;
        }
        if (target < arr[mid]) {
            //左侧找
            endIndex = mid - 1;
        } else {
            //右侧找
            startIndex = mid + 1;
        }
        return binarySearchRecursion(arr, target, startIndex, endIndex);
    }

七、二分查找在Java中的实现

Arrays.binarySearch(int[] a, int key)

    /**
     * Searches the specified array of ints for the specified value using the
     * binary search algorithm.  The array must be sorted (as
     * by the {@link #sort(int[])} method) prior to making this call.  If it
     * is not sorted, the results are undefined.  If the array contains
     * multiple elements with the specified value, there is no guarantee which
     * one will be found.
     *
     * @param a the array to be searched
     * @param key the value to be searched for
     * @return index of the search key, if it is contained in the array;
     *         otherwise, <tt>(-(<i>insertion point</i>) - 1)</tt>.  The
     *         <i>insertion point</i> is defined as the point at which the
     *         key would be inserted into the array: the index of the first
     *         element greater than the key, or <tt>a.length</tt> if all
     *         elements in the array are less than the specified key.  Note
     *         that this guarantees that the return value will be &gt;= 0 if
     *         and only if the key is found.
     */
    public static int binarySearch(int[] a, int key) {
        return binarySearch0(a, 0, a.length, key);
    }

    // Like public version, but without range checks.
    private static int binarySearch0(int[] a, int fromIndex, int toIndex,
                                     int key) {
        int low = fromIndex;
        int high = toIndex - 1;

        while (low <= high) {
            int mid = (low + high) >>> 1;
            int midVal = a[mid];

            if (midVal < key)
                low = mid + 1;
            else if (midVal > key)
                high = mid - 1;
            else
                return mid; // key found
        }
        return -(low + 1);  // key not found.
    }

八、二分查找对重复元素的处理

查找目标元素,如果有相同的元素,返回最左侧/最右侧的元素

    /**
     * 二分查找,返回最左侧的目标元素
     * @return
     */
    public static int binarySearchLeftMost(int[] arr, int target) {
        int i = 0;
        int j = arr.length - 1;
        int candidate = -1 ;
        while (i <= j) {
            int mid = (i + j) >>> 1;
            if (target < arr[mid]) {
                // 去到左区间查找
                j = mid - 1;
            } else if (target > arr[mid]) {
                // 去到右区间查找
                i = mid + 1;
            } else {
                //相等了不立刻返回,而是找到最左侧的,再返回
                candidate = mid;
                //继续向左查找,看看是否还有相同的
                j = mid - 1;
            }
        }
        return candidate;
    }

    /**
     * 二分查找,返回最右侧的目标元素
     * @return
     */
    public static int binarySearchRightMost(int[] arr, int target) {
        int i = 0;
        int j = arr.length - 1;
        int candidate = -1 ;
        while (i <= j) {
            int mid = (i + j) >>> 1;
            if (target < arr[mid]) {
                // 去到左区间查找
                j = mid - 1;
            } else if (target > arr[mid]) {
                // 去到右区间查找
                i = mid + 1;
            } else {
                //相等了不立刻返回,而是找到最右侧的,再返回
                candidate = mid;
                //继续向左查找,看看是否还有相同的
                i = mid + 1;
            }
        }
        return candidate;
    }

优化-1返回值:让找不到的时候,返回一个接近的元素下标,而不是-1

    /**
     * 优化最左查找返回值
     * @param arr
     * @param target
     * @return 返回大于等于目标的最靠左的索引
     */
    public static int binarySearchLeftMost1(int[] arr, int target) {
        int i = 0;
        int j = arr.length - 1;
        while (i <= j) {
            int mid = (i + j) >>> 1;
            if (target <= arr[mid]) {
                // 去到左区间查找
                j = mid - 1;
            } else {
                // 去到右区间查找
                i = mid + 1;
            }
        }
        return i;
    }


    /**
     * 优化最右查找返回值
     * @param arr
     * @param target
     * @return 小于等于目标的最靠右的索引
     */
    public static int binarySearchRightMost1(int[] arr, int target) {
        int i = 0;
        int j = arr.length - 1;
        while (i <= j) {
            int mid = (i + j) >>> 1;
            if (target < arr[mid]) {
                // 去到左区间查找
                j = mid - 1;
            } else {
                // 去到右区间查找
                i = mid + 1;
            }
        }
        return i - 1;
    }

九、Leetcode相关题目:

1.Leetcode704.二分查找

给定一个 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
class Solution {
    public int search(int[] nums, int target) {
        int i = 0;
        int j = nums.length -1 ;
        while(i<=j){
            int mid = (i+j) >>> 1;
            if(target < nums[mid]){
                j = mid -1;
            }else if(target > nums[mid]){
                i = mid +1;
            }else{
                return mid;
            }
        }
        return -1;
    }
}

2.Leetcode35. 搜索插入位置

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
class Solution {
    public int searchInsert(int[] nums, int target) {
        int i = 0;
        int j = nums.length-1;
        while(i<=j){
            int mid = (i+j) >>> 1;
            if(target<=nums[mid]){
                j = mid -1;
            }else {
                i = mid +1;
            }
        }
        return i;
    }
}

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

给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

class Solution {
    public int[] searchRange(int[] nums, int target) {
        // 查找最左侧元素
        int left = searchMost(nums,target,-1);
        if(left == -1){
            return new int[]{-1,-1};
        }else{
            // 查找最右侧元素
            int right = searchMost(nums,target,1);
            return new int[]{left,right};
        }
    }

    
    public int searchMost(int[] nums,int target,int lr){
        int i = 0;
        int j = nums.length - 1 ;
        int candidate = -1;
        while(i<=j){
            int mid = (i+j) >>> 1;
            if(target < nums[mid]){
                j = mid -1;
            }else if(target > nums[mid]){
                i = mid + 1;
            }else {
                candidate = mid;
                // 最左元素
                if(lr == -1){
                    j = mid - 1;
                }else{
                    //最右元素
                    i = mid + 1;
                }
            }
        }
        return candidate;
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342

推荐阅读更多精彩内容