在排序数组中查找元素的第一个和最后一个位置-二分查找34-python&c++

没看答案,考察的是有重复数字数组的二分查找的左右边界问题。
python

class Solution:
    def left_bound(self, nums, target):
        left, right = 0, len(nums)-1

        while left <= right:
            mid = int(left + (right - left)/2)

            if nums[mid] == target:
                right = mid - 1
            elif nums[mid] < target:
                left = mid + 1
            elif nums[mid] > target:
                right = mid - 1
        
        if left >= len(nums) or nums[left] != target:
            return -1
        
        return left
    
    def right_bound(self, nums, target):
        left, right = 0, len(nums)-1
        
        while left <= right:
            mid = int(left + (right - left)/2)

            if nums[mid] == target:
                left = mid + 1
            elif nums[mid] < target:
                left = mid + 1
            elif nums[mid] > target:
                right = mid - 1
        
        if right < 0 or nums[right] != target:
            return -1

        return right
    
    def searchRange(self, nums: List[int], target: int):
        res = []
        res.append(self.left_bound(nums, target))
        res.append(self.right_bound(nums, target))

        return res

c++

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int left_index = left_bound(nums, target);
        int right_index = right_bound(nums, target);
        return {left_index, right_index};
    }

    int left_bound(vector<int>& nums, int& target) {
        // 找左边界
        int left = 0, right = nums.size() - 1;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] == target) {
                right = mid - 1;
            } else if (nums[mid] > target) {
                right = mid - 1;
            } else if (nums[mid] < target) {
                left = mid + 1;
            }
        }

        if (left < 0 || left >= nums.size()) return -1;
        return nums[left] == target ? left : -1;
    }

    int right_bound(vector<int>& nums, int& target) {
        // 找右边界
        int left = 0, right = nums.size() - 1;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] == target) {
                left = mid + 1;
            } else if (nums[mid] > target) {
                right = mid - 1;
            } else if (nums[mid] < target) {
                left = mid + 1;
            }
        }

        if (right < 0 || right >= nums.size()) return -1;
        return nums[right] == target ? right : -1;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值