Search for a Range

最后更新于:2022-04-02 01:08:54

# Search for a Range ### Source - leetcode: [Search for a Range | LeetCode OJ](https://leetcode.com/problems/search-for-a-range/) - lintcode: [(61) Search for a Range](http://www.lintcode.com/en/problem/search-for-a-range/) ### Problem Given a sorted array of *n* integers, find the starting and ending position ofa given target value. If the target is not found in the array, return `[-1, -1]`. #### Example Given `[5, 7, 7, 8, 8, 10]` and target value `8`, return `[3, 4]`. #### Challenge O(log *n*) time. ### 题解 lower/upper bound 的结合,做两次搜索即可。 ### Java ~~~ public class Solution { /** *@param A : an integer sorted array *@param target : an integer to be inserted *return : a list of length 2, [index1, index2] */ public int[] searchRange(int[] A, int target) { int[] result = new int[]{-1, -1}; if (A == null || A.length == 0) return result; int lb = -1, ub = A.length; // lower bound while (lb + 1 < ub) { int mid = lb + (ub - lb) / 2; if (A[mid] < target) { lb = mid; } else { ub = mid; } } // whether A[lb + 1] == target, check lb + 1 first if ((lb + 1 < A.length) && (A[lb + 1] == target)) { result[0] = lb + 1; } else { result[0] = -1; result[1] = -1; // target is not in the array return result; } // upper bound, since ub >= lb, we do not reset lb ub = A.length; while (lb + 1 < ub) { int mid = lb + (ub - lb) / 2; if (A[mid] > target) { ub = mid; } else { lb = mid; } } // target must exist in the array result[1] = ub - 1; return result; } } ~~~ ### 源码分析 1. 首先对输入做异常处理,数组为空或者长度为0 1. 分 lower/upper bound 两次搜索,注意如果在 lower bound 阶段未找到目标值时,upper bound 也一定找不到。 1. 取`A[lb + 1]` 时一定要注意判断索引是否越界! ### 复杂度分析 两次二分搜索,时间复杂度仍为 O(logn)O(\log n)O(logn).
';