leetcode每日一题-34:在排序数组中查找元素的第一个和最后一个位置

最新推荐文章于 2024-08-02 09:03:52 发布

苦泉

最新推荐文章于 2024-08-02 09:03:52 发布

阅读量194

点赞数

分类专栏：社团算法打卡文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_45826803/article/details/120989344

版权

社团算法打卡专栏收录该内容

31 篇文章 13 订阅

订阅专栏

这篇博客介绍了如何使用二分查找解决LeetCode第34题，在已排序数组中查找元素的第一个和最后一个位置。通过两个整数二分模板，分别找到目标元素的左右边界，从而得到其在数组中的范围。代码提供了C++和Java两种实现方式。

摘要由CSDN通过智能技术生成

leetcode每日一题-34:在排序数组中查找元素的第一个和最后一个位置

链接

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

题目

在这里插入图片描述

分析

这个其实就是昨天题目的升级版本,掌握应用两种二分的模板即可,这两种模板可以应用在90%的整数二分问题当中

代码

class Solution {
public:
    // 两个整数二分模板的应用
    vector<int> searchRange(vector<int>& nums, int target) {
        int n = nums.size();
        vector<int> res;

        // 为空则一定无答案
        if(n == 0) return {-1, -1};

        // 找到第一个大于等于target的值
        int l = 0, r = n - 1;
        while(l < r)
        {
            int mid = l + r >> 1;
            if(nums[mid] >= target) r = mid;
            else l = mid + 1;
        }

        // 如果这个值等与target我们继续搜索taret的右边界
        if(nums[l] == target)
        {
            res.push_back(l);
            
            // 找到第一个小于等于target的值,注意mid值的变化
            l = 0, r = n-1;
            while(l < r)
            {
                int mid = l + r + 1 >> 1;
                if(nums[mid] <= target) l = mid;
                else r = mid - 1;
            }
            // cout << l << endl;
            res.push_back(l);
            return res;
        }
        else return {-1, -1};
    }
};

java

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int leftIdx = binarySearch(nums, target, true);
        int rightIdx = binarySearch(nums, target, false) - 1;
        if (leftIdx <= rightIdx && rightIdx < nums.length && nums[leftIdx] == target && nums[rightIdx] == target) {
            return new int[]{leftIdx, rightIdx};
        } 
        return new int[]{-1, -1};
    }

    public int binarySearch(int[] nums, int target, boolean lower) {
        int left = 0, right = nums.length - 1, ans = nums.length;
        while (left <= right) {
            int mid = (left + right) / 2;
            if (nums[mid] > target || (lower && nums[mid] >= target)) {
                right = mid - 1;
                ans = mid;
            } else {
                left = mid + 1;
            }
        }
        return ans;
    }
}

作者：LeetCode-Solution