Lintcode 搜索区间

给定一个包含 n 个整数的排序数组，找出给定目标值 target 的起始和结束位置。

如果目标值不在数组中，则返回[-1, -1]

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出[5, 7, 7, 8, 8, 10]和目标值target=8,

返回[3, 4]

挑战 

时间复杂度 O(log n)

标签 

二分法  排序数组  数组

分析：用二分法进行搜索，当搜索到之后，再用i，j分别向前、后查找target的最前端和最后端。

class Solution {
    /** 
     *@param A : an integer sorted array
     *@param target :  an integer to be inserted
     *return : a list of length 2, [index1, index2]
     */
public:
    vector<int> searchRange(vector<int> &A, int target) {
        // write your code here
        vector<int> result;
        result.push_back(-1);
        result.push_back(-1);
        if(A.empty())
        return result;
        int n=A.size();
        int begin=0,end=n-1,mid;
        
        while(begin<end){
            mid=(begin+end)/2;
            if(A[mid]==target){
              begin=mid;
              break;
            }
            else if(A[mid]<target)
            begin=mid+1;
            else
            end=mid-1;
        }
        int i,j;
        if(A[begin]==target){
            i=begin;
            j=begin;
            while(A[i]==target) i--;
            while(A[j]==target) j++;
            result[0]=i+1;
            result[1]=j-1;
        }
        
        return result;
        
    }
};

如有问题请留言。

转载请注明出处。