leetcode hot 100-34. 在排序数组中查找元素的第一个和最后一个位置

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

给定一个按照升序排列的整数数组 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]

思路:二分法

这题和 leetcode 剑指 Offer 53 - I. 在排序数组中查找数字 I 的实现思路很像,都是查找 taret 的左右边界,

在二分查找法中使用 if(target >= nums[i])  left = mid + 1;可以找到第一个大于target的元素的位置。

其中该方法查找 target 的右边界,复用该函数查找 (target-1)的右边界即是 target 的左边界,右边界所指元素是第一个大于target元素,左边界所指元素是第一个大于 (target-1)的元素,

  • 如果target存在的话,则因为左边界指向target的首次出现的值,右边界指向的是第一个大于 target 的元素的位置,所以target的位置范围为 [leftBorder, rightBorder - 1];
  • 如果 target 不存在的话,右边界指向的是第一个大于target 的元素的位置,左边界指向的是第一个大于 (target -1) 的元素,本来应该指向target的,当时因为target 不存在,所以也指向了第一个大于target 的元素,所以此时 左右边界相等,leftBorder == rightBorder。
 1 class Solution {
 2 
 3     // 获取在nums数组第一个大于等于 target 的元素下标
 4     public int getRightBorder(int[] nums, int target){
 5         int left = 0, right = nums.length - 1;
 6         int mid = 0;
 7         while(left <= right){
 8             mid = (left + right) / 2;
 9             if(target >= nums[mid]){
10                 left = mid  + 1;
11             }else{
12                 right = mid - 1;
13             }
14         }
15         return left;
16     }
17 
18     public int[] searchRange(int[] nums, int target) {
19         if(nums == null || nums.length == 0){
20             return new int[]{-1, -1};
21         }
22 
23         int leftBorder = getRightBorder(nums, target - 1);  
24         int rightBorder = getRightBorder(nums, target);
25         if(leftBorder == rightBorder){        // 左右下标相等则说明该数不存在
26             return new int[]{-1, -1};
27         }
28         return new int[]{leftBorder , rightBorder - 1};
29     }
30 }
leetcode 执行用时:0 ms > 100.00%, 内存消耗:41.9 MB > 69.63%

复杂度分析:

时间复杂度:O(logn)。使用二分法查找了 target  的左右边界,所以时间复杂度为O(logn)。

空间复杂度:O(1)。只使用了常数个大小的变量空间,所以空间复杂度为O(n)。

你可能感兴趣的:(二分法,算法,数据结构,leetcode,快速排序)