leetcode -- Search for a Range

最新推荐文章于 2018-09-05 14:58:19 发布

原创最新推荐文章于 2018-09-05 14:58:19 发布 · 699 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #binary search

数据结构算法专栏收录该内容

23 篇文章

订阅专栏

本文介绍了如何通过二分查找算法在有序数组中高效地找到给定目标值的起始和结束位置。对于未找到目标值的情况，返回[-1, -1]。详细步骤包括递归出口条件以及处理目标值大于中间元素的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Q:Given a sorted array of integers, find the starting and ending position of a given target value.

Your algorithm's runtime complexity must be in the order of O(log n).

If the target is not found in the array, return [-1, -1].

For example, Given [5, 7, 7, 8, 8, 10] and target value 8, return [3, 4].

分析:因为是有顺序序列,应该可以通过二分法查找.两个递归出口:(1)n=1且没有目标找到;(2)已经找到目标.
需要注意的是target>A[n/2]的情况,这时对[A+n/2, n-n/2]子序列重新递归查找,返回值target位置要加上之前的n/2.

class Solution {
public:
    vector<int> searchRange(int A[], int n, int target) {
        vector<int> ret;
        if(n == 1 && target !=A[0])   //递归出口1,找不到目标值
            ret.assign(2, -1);
        else if(target == A[n/2]){      //递归出口2,找到目标值
            int tbeg = n/2, tend = n/2;
            while((--tbeg) >= 0 && target == A[tbeg]);
            while((++tend) < n && target == A[tend]);
            ret.push_back(tbeg + 1);
            ret.push_back(tend - 1);
        }
        else if(target < A[n/2])
            ret = searchRange(A, n/2, target);
        else{
	    ret = searchRange(A + n/2, n - n/2, target);
	    if(ret[0] == -1 && ret[1] == -1)  return ret;
	    ret[0] += n/2; ret[1] += n/2;       //***attention***
        }
        return ret;
    }
};