【LeetCode】34. Search for a Range

题目描述

Given an array of integers sorted in ascending order, 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].

解题思路

Binary Search。
题目要求O(log n)的时间复杂度，所以考虑使用二分查找。由于需要找整个范围的前后两个界限，所以考虑使用两次二分查找分别来找左右两个数字。
对于左边的界限，每次检索，考虑如下三种情况:

1.  target > nums[mid], 那么往右继续查找
2.  target < nums[mid], 那么往左继续查找
3.  targe == nums[mid], 那么它有可能是左边界限，也有可能左界还在它的前面，所以继续往左找

这样，后两种情况可以合并。最后判断leftIdx停留在的位置是否为target，如果不是，说明target不存在数组中，也就不会有什么右界，直接return。否则，继续查找右界。
现在左界已经确定，只需从[leftIdx, nums.size()) 这个范围中查找右界。如同上面的分析，我们可以知道在找右界时，每次检索，有如下情况

1. target >= nums[mid], 往右查找
2. target < nums[mid], 往左查找

注意，由于int除法总是往下round，所以我们的midIdx永远是偏向于左边的，而当我们发现情况1时，会选择更新midIdx = leftIdx。那么，问题来了，如果出现了如下情况
[5, 8], target=5
算法就会陷入死循环，所以我们应该设法让midIdx右偏，让midIdx=(leftIdx+rightIdx)/2+1，这是非常关键的。
在二分查找中，考虑如下两种情况：

1. 如果要使找到的值在最前面，那么，对于target==nums[mid],应该往左找。所以对于往右的情况，不需要在考虑中间值，只需设置leftIdx=nums[mid]+1，这样也不会受round左偏的影响。
2. 如果要使找打的值在最后面，那么，对于target==nums[mid]，应该往右找。由于在往右时，我们还是要考虑中间值的情况，所以设置leftIdx=nums[mid]。这就会出现上述的mid左偏死循环的情况，所以要给mid+1。

AC代码

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ans;
        ans.push_back(-1);
        ans.push_back(-1);
        if (nums.empty())
            return ans;

        const int vSize = nums.size();

        //find the start index 
        int leftIdx = 0, rightIdx = vSize - 1;
        int midIdx; 
        while (leftIdx < rightIdx) {
            midIdx = (leftIdx + rightIdx) / 2;
            if (target > nums[midIdx]) {
                leftIdx = midIdx + 1;
            }
            else {
                rightIdx = midIdx;
            }
        }
        if (nums[leftIdx] == target)
            ans[0] = leftIdx;
        else 
            return ans;

        //find the end index, start from the leftIdx found above 
        rightIdx = vSize - 1;
        while (leftIdx < rightIdx) {
            midIdx = (leftIdx + rightIdx) / 2 + 1; // Make mid bais to right
            if (target < nums[midIdx]) 
                rightIdx = midIdx - 1;
            else 
                leftIdx = midIdx;
        }
        ans[1] = leftIdx;

        return ans;
    }
};