JAVA程序设计:在排序数组中查找元素的第一个和最后一个位置(LeetCode:34)

第一次时间耗时超过了100%的用户,纪念一下哈哈哈哈

JAVA程序设计:在排序数组中查找元素的第一个和最后一个位置(LeetCode:34)_第1张图片

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

你的算法时间复杂度必须是 O(log n) 级别。

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

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]

思路:两次二分即可,第一次一直让目标索引一直往左,第二次一直往右即可。。

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] ans=new int[2];
        ans[0]=ans[1]=-1;
        if(nums.length==0) return ans;
        if(nums.length==1)
        {
        	if(nums[0]==target)
        		ans[0]=ans[1]=0;
        	return ans;
        }
        int l=0,r=nums.length-1,p=-1;
        while(l<=r)
        {
        	int mid=(l+r)/2;
        	if(nums[mid]>=target)
        	{
        		p=mid;
        		r=mid-1;
        	}
        	else
        		l=mid+1;
        }
        if(p==-1 || nums[p]!=target) return ans;
        ans[0]=p;l=p;r=nums.length-1;
        while(l<=r)
        {
        	int mid=(l+r)/2;
        	if(nums[mid]>target)
        		r=mid-1;
        	else
        	{
        		p=mid;
        		l=mid+1;
        	}
        }
        ans[1]=p;
        return ans;
    }
}

 

你可能感兴趣的:(JAVA程序设计:在排序数组中查找元素的第一个和最后一个位置(LeetCode:34))