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

tags: array, binary search

简单题的变形,需要考虑找到结果后,从结果位置出发向左向右搜索所有与结果连续相等的数字。

中间有一点需要注意:left和right表示的是第一个不符合条件的元素位置,我们需要对left加一,对right减一,使之满足我们的条件。

class Solution {
public:
    vector<int> searchRange(int A[], int n, int target) {
        vector<int> re;
        _searchRange(re, A, 0, n, target);
        return re;
    }
    void _searchRange(vector<int> &result, int A[], int start, int end, int target){
        int mid = (start + end) / 2;
        if(start + 1 > end){//not found
            result.push_back(-1);
            result.push_back(-1);
            return;
        }
        else if(A[mid] == target){//found
            int left = mid;
            int right = mid;
            while(A[left] == target && left >= start){
                left--;
            }
            left++;//important
            while(A[right] == target && right < end){
                right++;
            }
            right--;//important
            result.push_back(left);
            result.push_back(right);
            return;
        }
        else{
            if(A[mid] > target){
                _searchRange(result, A, start, mid, target);
            }
            else{
                _searchRange(result, A, mid + 1, end, target);
            }
        }
        
    }
};

Selection_005

 

 

 

 

Leave a comment

Your email address will not be published. Required fields are marked *

This site uses Akismet to reduce spam. Learn how your comment data is processed.