二分查找

如果面试时面试官让你写一个二分查找你会怎么写呢?不少同学可能都会这样写

int binary_search(const vector<int> &nums, int target) {
    int l = 0, r = nums.size();
    while (l < r) {
        int mid = l + (r - l) / 2;
        if (target == nums[mid])
            return mid;
        else if (target < nums[mid])
            r = mid;
        else
            l = mid + 1;
    }
    return -1;
}

如果面试的时候这样写了,并且其他方面又没有什么亮点,很不幸,那可能就要回去等通知了。

首先,这段代码确实没有错;但是它有一个缺陷,就是它返回的下标并不是确定的。什么意思呢?如果需要查找的target在有序数组中存在多个,我们没法确定它到底会返回哪一个。
例如:

target: 2
Input: [1, 2, 2, 3, 4]
Output: 2

Input: [1, 2, 2, 2, 2, 2, 2, 3, 4]
Output: 4

对于上面的两个数组一个返回了2,另一个返回了4,结果确实没有错。但如果函数能返回找到的第一个target的下标那肯定就更好了,这样对于两个输入函数都应该会返回1。

那么需要怎样修改才能满足上面的要求呢?首先大体框架不会发生变化,需要变化的是一些细节;当然,有序数组还是需要保证的,这里使用非逆序数组来讲解。并且我们使用左闭右开区间 [l,r),即包含左侧元素而不包含右侧元素。

  • 当 target==nums[mid] 时,不要立刻返回下标。而是令 r=mid,这么做的目的是让搜索区间往左边挪动,挪到最后 l 和 r 相等,此时 l 就是最左侧的 target 下标。

  • 当 target<nums[mid] 时,令 r=mid,因为 nums[mid] 已经比 target 大了,nums[mid] 以及右侧的数都不可能等于 target。注意搜索区间是不包含 nums[r] 的。

  • 当 target>nums[mid] 时,令 l=mid+1,因为 nums[mid] 已经比 target 小了,nums[mid] 左侧的数都不可能等于 target,至于为什么这里需要加1,是因为使用的是左闭右开区间。

int binary_search(const vector<int> &nums, int target) {
    int l = 0, r = nums.size();
    while (l < r) {
        int mid = l + (r - l) / 2;
        if (target <= nums[mid])
            r = mid;
        else
            l = mid + 1;
    }
    if ((l < 0 || nums.size() <= l) || nums[l] != target)
        return -1;
    return l;
}

根据上面的方法,在退出循环时 l==r。如果数组中存在 target,那么 l 就是最左侧 target 的下标;如果数组中不存在 target,那么 l 就是将 target 插入到数组中并且保持数组有序时 target 的下标。

数组中存在 target 的情形,最终 l 和 r 会在最左侧的 target 相遇 (r=mid 会让区间不断地向最左侧的 target 缩小),然后退出循环。

数组中不存在 target 的情形

  • target 小于数组中所有的数,则不断使 r=mid,最终使得 l==r,即退出循环。
  • target 大于数组中所以的数,则不断使 l=mid+1,最终使得 l==r。
  • target 位于数组的区间范围时,假设对于 a<target<b,且数组 nums 中存在 a,b。如果 nums[mid]<target,l 需要不停地向右移动,并且至多移动到 b 的位置上;同理如果 target<=nums[mid],r 需要向左移动,并且至多移动到 b 的位置上。最终 l 和 r 相遇并退出循环。

综上分析,实际上循环结束时 l 的值就是假设数组中存在 target 时的合理下标。最后我们对于没有找到 target 时做特殊处理,返回-1。

上面我们已经知道了找左侧边界的二分查找怎样实现,那么找右侧边界的二分查找就很简单了,原理是相通的。

int binary_search(const vector<int> &nums, int target) {
    int l = 0, r = nums.size();
    while (l < r) {
        int mid = l + (r - l) / 2;
        if (nums[mid] <= target)
            l = mid + 1;
        else
            r = mid;
    }
    if ((r - 1 < 0 || nums.size() <= r - 1) || nums[r - 1] != target)
        return -1;
    return r - 1;
}

如果数组中存在 target ,最终 r 也是停在最右侧 target 的右边 (因为 r 向左移动的条件是 target < nums[mid]),所以退出循环后 r 需要减1才是最右侧 target 的下标。这样一来,当数组中不存在 target 时,r-1 就可以定义为“在数组中为target找一个可以使得数组仍然保持有序的位置的前一个下标”。

看到这里,相信各位已经彻底掌握了二分查找,面试的时候不会再因为写不出让面试官满意的二分查找而喜提感谢信。

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

推荐阅读更多精彩内容

  • ------ 二分查找基本版及其各种变形的汇总 思想: 二分搜索的核心就是**循环结束条件**和**左右边界迭代规...
    xushichao阅读 269评论 0 0
  • 二分查找法 说明:二分查找法在代码实现上有模板方法,一定要掌握。 1、二分查找法的使用前提:数组一定要是排好序的,...
    李威威阅读 685评论 0 1
  • 基于有序数组的二分查找法 查找问题是计算机中非常重要的一类基础问题,也是我们生活中常见的问题。 在我们的生活中,要...
    李威威阅读 544评论 0 0
  • 终结条件为根据if (check(mid)) 条件成立,判断答案在左区间还是右区间,如果答案在左区间并且mid也可...
    舒也ella阅读 648评论 0 0
  • LeetCode 第 704 题是二分查找的模板问题。 LeetCode 第 704 题:二分查找 传送门:704...
    李威威阅读 1,709评论 0 1