Search for a Range

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].

思路并不复杂,就是先用二分查找找到其中一个target,然后再往左右找到target的边缘。找边缘的方法跟二分查找仍然是一样的,只是切半的条件变成相等,或者不等(往左边找则是小于,往右边找则是大于)。或者 如果我们不寻找那个元素先,而是直接相等的时候也向一个方向继续夹逼,如果向右夹逼,最后就会停在右边界,而向左夹逼则会停在左边界,如此用停下来的两个边界就可以知道结果了,只需要两次二分查找。

C++实现代码:

#include<iostream>
#include<vector>
#include<set>
using namespace std;

class Solution
{
public:
    vector<int> searchRange(int A[], int n, int target)
    {
        if(n==0)
            return vector<int>{-1,-1};
        int mid;
        int left=0;
        int right=n-1;
        int ll=-1;
        int rr=-1;
        while(left<=right)
        {
            mid=(left+right)/2;
            if(A[mid]<=target)
            {
                left=mid+1;
            }
            else
                right=mid-1;
        }
        if(A[right]==target)
            rr=right;
        left=0;
        right=n-1;
        while(left<=right)
        {
            mid=(left+right)/2;
            if(A[mid]>=target)
                right=mid-1;
            else
                left=mid+1;
        }
        if(A[left]==target)
            ll=left;
        return vector<int>{ll,rr};
    }
};

int main()
{
    Solution s;
    int arr[6]={5};
    vector<int> result=s.searchRange(arr,1,1);
    for(auto a:result)
        cout<<a<<" ";
    cout<<endl;
}
原文地址:https://www.cnblogs.com/wuchanming/p/4130178.html