LeetCode 之

来源:互联网 发布:盐城java软件培训 编辑:程序博客网 时间:2024/03/29 12:54

题目链接:Search for a Range
对于一个按照升序排序的整数数组,找到给定目标值的开始位置和结束位置,且时间复杂度为O(log n)。
如果在数组中找不到目标值,则返回[-1,-1]。

例如:
数组为[5,7,7,8,8,10]和目标值8;
返回[3,4]。

首先如果使用暴力查找,算法的时间复杂度绝对会超过O(log n),所以不予考虑。
解法一:使用双指针,i指向数组头,j指向数组尾,分别进行查找,找到目标值的开始和结束位置。
代码如下:

class Solution {    public int[] searchRange(int[] nums, int target) {        int len = nums.length;        int i = 0, j = len-1;        while(i<=j&&(nums[i]!=target||nums[j]!=target)){            if(nums[i]!=target)                i++;            if(nums[j]!=target)                j--;        }        int[] re = {i,j};        if(i>j){            re[0] = -1;            re[1] = -1;           }        return re;    }}

解法二:使用二分法
通过二分法找到target的位置,不管起是否为target的起始或者结束位置,找到后,向左寻找起始位置,向右寻找结束位置;
其中需要注意的是,当由二分法找到的target的位置为0或者为len-1时需要注意越界。
代码如下:

class Solution {    public int[] searchRange(int[] nums, int target) {        int len = nums.length;        int tmp = binfind(nums,target);        int[] re = {-1,-1};         if(tmp==-1){            return re;        }        int i = tmp-1, j = tmp+1;        while(i>-1&&nums[i]==target)            i--;        while(j<len&&nums[j]==target)            j++;        re[0] = i+1;        re[1] = j-1;        return re;    }   public int binfind(int[] nums, int target){            int len = nums.length;            int i = 0, j = len-1;            while(i<=j){                int mid = (i+j)/2;                if(nums[mid]>target){                    j = mid-1;                }                else {                    if(nums[mid]<target){                        i = mid+1;                    }                    else {                          return mid;                    }                    }                }              return -1;        }}
原创粉丝点击