35. 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0
代码实现 |
class Solution {
public int searchInsert(int[] nums, int target) {
int i =0;
while(i<nums.length){
if(nums[i]>=target)break;
i++;
}
return i;
}
}
官方解法 |
class Solution35 {
public int searchInsert(int[] nums, int target) {
int n = nums.length;
int left = 0, right = n - 1, ans = n;
while (left <= right) {
int mid = ((right - left) >> 1) + left;
if (target <= nums[mid]) {
ans = mid;
right = mid - 1;
} else {
left = mid + 1;
}
}
return ans;
}
}
//链接:https://leetcode-cn.com/problems/search-insert-position/solution/sou-suo-cha-ru-wei-zhi-by-leetcode-solution/
官方是使用了二分法,在一个有序的数组中,使用二分查找target的位置的时间复杂度为NlogN。
观察示例数据和题意,我们知道target是
第一种情况 存在这个值 大于左和小于右 即为第一次num[i]==target的索引位置
第二种情况 不存在这个值 插入位置即为第一次num[i]>target的索引位置
二分查找法直接根据left<target<right的特性,利用mid节省比较次数