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

题目:

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

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

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

示例:
在这里插入图片描述
代码:

  • 解法一
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int i = 0, 
        int j = nums.length;
        int mid = (i + j) / 2;
        int p = -1;
        while (i < j) {
            if (nums[mid] == target) {
                p = mid;
                break;
            }
            if (nums[mid] > target) {
                if (j == mid)
                 break;
                j = mid;
                mid = (i + j) / 2;
            } else {
                if (i == mid) break;
                i = mid;
                mid = (i + j) / 2;
            }
        }
        if (p == -1) {
            return new int[]{-1, -1};
        } else {
            int a = p, b = p;
            while (a > 0 && nums[a - 1] == target) {
            	a--;
            }
            while (b < nums.length - 1 && nums[b + 1] == target){
           		b++;
             }
            return new int[]{a, b};
        }
    }
}
  • 别人的代码
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] res = new int[2]; 
        int l = 0;
        int r = nums.length-1;
        int mid = (l + r)/2;
        boolean flag = false;
        while(l <= r){
            mid = (l + r)/2;
            if(nums[mid] > target){
                r = mid - 1;
            }else if(nums[mid] < target){
                l = mid + 1;
            }else if(nums[mid] == target){
                flag = true;
                break;
            }
        }  
        if(flag == false){
            res[0] = -1;
            res[1] = -1;
        }else{
            int i = mid-1;
            int j = mid+1;
            while(i>=0 && nums[i] == target){     //while循环条件里的i>=0与nums[i] == target的顺序,
            顺序不同导致两个条件判断先后顺序不同,谁在前就先判断哪个条件;如果调换则会造成数组下标越界
                i--;
            }
            res[0] = i+1;
            
            while(j<nums.length && nums[j] == target){
                j++;
            }
            res[1] = j-1;
        }
        
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/inferno_devil/article/details/88686289