前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >在排序数组中查找元素的第一个和最后一个位置(leetcode34)

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

作者头像
Vincent-yuan
发布2021-04-09 15:14:02
1.8K0
发布2021-04-09 15:14:02
举报
文章被收录于专栏:Vincent-yuan

给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target,返回 [-1, -1]。

进阶:

你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?

示例 1:

代码语言:javascript
复制
输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]

解析:

方法一:二分查找

二分查找中,寻找leftIdx 即为在数组中寻找第一个大于等于 target 的下标,寻找 rightIdx 即为在数组中寻找第一个大于target 的下标,然后将下标减一。

两者的判断条件不同,为了代码的复用,我们定义 binarySearch(nums, target, lower) 表示在 nums 数组中二分查找 target 的位置,如果 lower 为 true,则查找第一个大于等于 target 的下标,否则查找第一个大于target 的下标。

代码实现

代码语言:javascript
复制
public class leetcode34 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub

        int[] nums = new int[]{5,7,7,8,8,10};
        int target = 8;
        
        int[] result = searchRange(nums,target);
        System.out.println(result[0]+","+result[1]);
    }
    
    public static int[] searchRange(int[] nums, int target) {

        int leftIdx = searchBinary(nums,target,true);
        int rightIdx = searchBinary(nums,target,false)-1;
        if(rightIdx<nums.length&&leftIdx>=0&&leftIdx<=rightIdx&&nums[leftIdx]==target&&nums[rightIdx]==target){
            return new int[]{leftIdx,rightIdx};
        }
        return new int[]{-1,-1};
    }
    
    public static int searchBinary(int[] nums,int target,boolean lower){
        int ans = nums.length;
        int left = 0;
        int right = nums.length-1;
        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;
    }

}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021-03-24 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档