题目描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
思路: 给定数组查找指定元素值的索引,如果元素值不存在于数组,就返回被顺序插入位置,并且时间复杂度要求O(log n),那么很自然就能想到使用二分查找,当二分查找找不到元素值时后面再去考虑顺序插入的情况。如果原数组不包括给定的元素值那么就要寻找插入位置,在我看来插入位置分三种情况,第一种指定元素小于等于数组中第一个数,第二种指定元素值大于等于数组最后一个数,第三种就是大于等于前一个数,小于等于后一个数。具体看代码。
代码:
class Solution {public int searchInsert(int[] nums, int target) {int left=0;int right=nums.length-1;while(left<=right) {int mid=(right+left)/2;if(nums[mid]<target) {left=mid+1;} else if(nums[mid]>target) {right=mid-1;} else {return mid;}}for(int i=0;i<nums.length-1;i++) {if(nums[i]<=target&&nums[i+1]>=target) {return i+1;} }if(nums[nums.length-1]<=target) {return nums.length;}return 0;}
}