【LeetCode】 34. Find First and Last Position of Element in Sorted Array 在排序数组中查找元素的第一个和最后一个位置(Medium)(JAVA)
题目地址: https://leetcode.com/problems/find-first-and-last-position-of-element-in-sorted-array/
题目描述:
Given an array of integers nums sorted in ascending order, find the starting and ending position of a given target value.
Your algorithm’s runtime complexity must be in the order of O(log n).
If the target is not found in the array, return [-1, -1].
Example 1:
Input: nums = [5,7,7,8,8,10], target = 8
Output: [3,4]
Example 2:
Input: nums = [5,7,7,8,8,10], target = 6
Output: [-1,-1]
题目大意
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
解题方法
1、看到你的算法时间复杂度必须是 O(log n) 级别,而且是有序数组,就知道要用二分查找
2、找到第一个位置
3、找到最后一个位置
note: 区分第一还是最后一个,就是通过 nums[mid] == target,放在 start 还是放在 end
class Solution {
public int[] searchRange(int[] nums, int target) {
int start = 0;
int end = nums.length - 1;
while (start <= end) {
int mid = start + (end - start) / 2;
if (nums[mid] >= target) {
end = mid - 1;
} else {
start = mid + 1;
}
}
int[] res = new int[]{-1, -1};
if (start >= nums.length || nums[start] != target) return res;
res[0] = start;
end = nums.length - 1;
while (start <= end) {
int mid = start + (end - start) / 2;
if (nums[mid] > target) {
end = mid - 1;
} else {
start = mid + 1;
}
}
res[1] = end;
return res;
}
}
执行用时 : 0 ms, 在所有 Java 提交中击败了 100.00% 的用户
内存消耗 : 42.8 MB, 在所有 Java 提交中击败了 40.86% 的用户