一题错过Facebook:二分查找+倍增
Posted LintCode领扣
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了一题错过Facebook:二分查找+倍增相关的知识,希望对你有一定的参考价值。
LintCode 447
在大数组中查找
如果你访问了一个不可访问的下标(比如越界),ArrayReader 会返回2,147,483,647。
输入: [1, 3, 6, 9, 21, ...], target = 3
输出: 1
输入: [1, 3, 6, 9, 21, ...], target = 4
输出: -1
解题思路
该题目有两种思路: 二分和倍增. 其实这两种方法有一些相似之处, 很多时候可以在相同的时空分析下解决同一个问题.
方法1 倍增:
首先特判一下首个元素. 然后设定 idx = 0 为查找的下标, jump = 1 为向后跳跃的长度.
每次循环将 idx 向后移动 jump 个元素, 并将 jump 翻倍. 而如果移动后的位置不小于 target, 则 jump 缩小至一半.
即我们在保证每次跳跃后的 idx 的位置都小于target的前提下, 倍增式地跳跃, 以此保证 O(logn) 的时间复杂度.
循环终止的条件就是 jump == 0, 就是说, 这时 idx + 1 的位置以及不小于 target 了 (此时idx位置的仍然是小于target)
也就是说, 到最后idx指向的元素是: 最大的小于target的元素. 返回答案前判断一下 idx + 1 是否 target 即可.
方法2 二分:
二分查找第一个不小于target的元素很简单. 但是需要确定二分区间的范围. 此时还是需要倍增地找到右边界.
初始右边界为1, 如果右边界的数小于 target, 就将其倍增, 直到右边界不小于target.
这时就可以二分查找了.
注意: 越界访问是没有关系的, 因为这个ArrayReader在越界访问时, 返回 INT_MAX, 一定不小于 target. 而即使是返回 -1 之类的数值, 我们也可以加一个判断搞定.
public class Solution {
/*
* @param reader: An instance of ArrayReader.
* @param target: An integer
* @return: An integer which is the first index of target.
*/
public int searchBigSortedArray(ArrayReader reader, int target) {
int firstElement = reader.get(0);
if (firstElement == target)
return 0;
else if (firstElement > target)
return -1;
int idx = 0, jump = 1;
while (jump != 0) {
while (jump != 0 && reader.get(idx + jump) >= target) // 越界时返回INT_MAX, 必然不小于target
jump >>= 1;
idx += jump;
jump <<= 1; // 当jump为0时, 左移一位不影响它的值, 不影响循环结束
}
if (reader.get(idx + 1) == target)
return idx + 1;
else
return -1;
}
}
/////////////// 方法2 二分
/**
* Definition of ArrayReader:
*
* public class ArrayReader {
* public int get(int index) {
* // return the number on given index,
* // return 2147483647 if the index is invalid.
* }
* };
*/
public class Solution {
/*
* @param reader: An instance of ArrayReader.
* @param target: An integer
* @return: An integer which is the first index of target.
*/
public int searchBigSortedArray(ArrayReader reader, int target) {
int l = 0, r = 1, mid;
while (reader.get(r) < target) // 越界返回INT_MAX, 必然大于target, 所以没有关系
r <<= 1;
while (l < r) {
mid = (l + r) >> 1;
if (reader.get(mid) >= target)
r = mid;
else
l = mid + 1;
}
if (reader.get(l) == target)
return l;
else
return -1;
}
}
以上是关于一题错过Facebook:二分查找+倍增的主要内容,如果未能解决你的问题,请参考以下文章