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

  1. 题目链接

  2. 解题思路:整体框架就是二分,只是找到了target后,还要继续查找。分两次查找,第一次是「找到等于target最左的位置」,第二次是「找到等于target最右的位置」

  3. 代码

    class Solution {
    public:
        vector<int> searchRange(vector<int>& nums, int target) {
            // 先找到target的最左的位置
            int L = 0;
            int R = nums.size() - 1;
            int max_left = -1;
            while(L <= R) {
                int mid = (R + L) / 2;
                if (nums[mid] == target) {
                    max_left = mid;
                    R = mid - 1;
                } else if (nums[mid] >= target) {
                    R = mid - 1;
                } else {
                    L = mid + 1;
                }
            }
            if(max_left == -1) {    // 没找到
                return {-1, -1};
            }
            int max_right = -1;
            L = 0;
            R = nums.size() - 1;
            while(L <= R) {
                int mid = (R + L) / 2;
                if (nums[mid] == target) {
                    max_right = mid;
                    L = mid + 1;
                } else if (nums[mid] >= target) {
                    R = mid - 1;
                } else {
                    L = mid + 1;
                }
            }
            return {max_left, max_right};
        }
    };
    
posted @ 2024-12-18 14:53  ouyangxx  阅读(7)  评论(0)    收藏  举报