leetcode34. 在排序数组中查找元素的第一个和最后一个位置

二分查找不用说,主要是这里注意如何找到数组中等于target的最左边和最右边的索引位置


image.png
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> res = {-1, -1};
        int rightIndex = findRightIndex(nums, target);
        if (rightIndex < 0 || nums[rightIndex] != target) return res;
        int leftIndex = findLeftIndex(nums, target);
        res[0] = leftIndex;
        res[1] = rightIndex;
        return res;
    }
    
    int findLeftIndex(vector<int>& nums, int target) {
        int lo = 0, hi = nums.size() - 1;
        while (lo <= hi) {
            int mid = lo + (hi - lo) / 2;
            if (nums[mid] < target) lo = mid + 1;
            else hi = mid - 1;
        }
        return lo;
    }
    
    int findRightIndex(vector<int>& nums, int target) {
          int lo = 0, hi = nums.size() - 1;
        while (lo <= hi) {
            int mid = lo + (hi - lo) / 2;
            if (nums[mid] > target) hi = mid - 1;
            else lo = mid + 1;
        }
        return hi;
    }
};
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容