【LeetCode】34. Search for a Range

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

由于O(logn)时间要求，显然用二分查找。

思路是先用二分查找找到其中一个target，找不到则返回默认的ret值[-1, -1]

找到之后从这个位置往两边递归进行二分查找进行范围的拓展。

具体来说，ret[0]不断向左扩展，ret[1]不断向右扩展。

class Solution {
public:
    vector<int> searchRange(int A[], int n, int target) {
        vector<int> ret(, -);
        int left;
        int right;
        int low = ;
        int high = n-;
        while((left = binarySearch(A, low, high, target)) != -)
        {
            ret[] = left;
            high = left-;
        }
        low = ;
        high = n-;
        while((right = binarySearch(A, low, high, target)) != -)
        {
            ret[] = right;
            low = right+;
        }
        return ret;
    }
    int binarySearch(int A[], int left, int right, int target)
    {
        while(left <= right)
        {
            int mid = left + (right-left) / ;
            if(A[mid] == target)
                return mid;
            else if(A[mid] < target)
                left = mid + ;
            else
                right = mid - ;
        }
        return -;
    }
};