[LeetCode] #35 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
输入: nums = [1,3,5,6], target = 5
输出: 2
重点在于O(log n),所以需要在一个循环解决问题
那么,我们逐个比较,当target等于数组元素时返回当前元素下标;小于时需要插入到当前元素前面,也就是当前元素的位置,所以也返回当前元素下标。
如果循环结束仍未返回,则说明target大于所有元素,需要插入到最后,所以返回数组的长度。
class Solution { public int searchInsert(int[] nums, int target) { for(int i = 0; i < nums.length; i++){ if(nums[i] >= target) return i; } return nums.length; } }
当然,对于有序序列的查找首选还是二分法
class Solution { public int searchInsert(int[] nums, int target) { int low = 0; int high = nums.length - 1; while (low <= high) { int mid = low + (high - low) / 2; if (nums[mid] > target) high = mid - 1; else if (nums[mid] < target) low = mid + 1; else return mid; } return high + 1; } }
知识点:无
总结:无