35. 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。(你可以假设数组中无重复元素)
输入: [1,3,5,6], 5
输出: 2
输入: [1,3,5,6], 2
输出: 1
输入: [1,3,5,6], 7
输出: 4
输入: [1,3,5,6], 0
输出: 0
public static int searchInsert(int[] nums, int target) {
int left = 0;
int right = nums.length - 1;
while(left <= right){
int mid = (left + right) / 2;
if(target<nums[mid]){
right = mid - 1;
}
else if(target == nums[mid]){
return mid;
}else{
left = mid + 1;
}
}
return left;
}
思路:
简单的二分查找模板。
34. 在排序数组中查找元素的第一个和最后一个位置
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。(你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?)
如果数组中不存在目标值 target,返回 [-1, -1]。
输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]
输入:nums = [5,7,7,8,8,10], target = 6
输出:[-1,-1]
输入:nums = [], target = 0
输出:[-1,-1]
class Solution {
public int[] searchRange(int[] nums, int target) {
if(nums.length == 0){
return new int[]{-1,-1};
}
int first = findFirstPostion(nums,target);
int last = findLastPostion(nums,target);
return new int[]{first,last};
}
public int findFirstPostion(int[] nums,int target){
int left = 0;
int right = nums.length - 1;
while(left < right){
//注意!!!
int mid = (left + right) / 2;
if(nums[mid] < target){
left = mid + 1;
}else if (target == nums[mid]){
//注意!!!
right = mid;
}else{
//nums[mid] > target
right = mid -1;
}
}
if(target == nums[left]){
return left;
}
return -1;
}
public int findLastPostion(int[] nums,int target){
int left = 0;
int right = nums.length -1;
while(left < right){
//注意!!!
int mid = (left + right + 1) / 2;
if(target > nums[mid]){
left = mid + 1;
}else if(target == nums[mid]){
//注意!!!
left = mid;
}else{
//target < nums[mid]
right = mid -1;
}
}
if(target == nums[right]){
return right;
}
return -1;
}
}
思路:
分两次查找,第一次查找第一个元素位置,第二次查找第二个元素位置。
有两点值得注意的:
- 寻找第一个元素位置方法里不会出现死循环,可以向下取整,但是在第二个方法里 出现了 left = mid + 1; 和 left = mid; ,可能会导致死循环,所以要 向上取整。
- 注意 if(target == nums[mid])的取数范围。