一、题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
二、思路
1、基础思路
从i=0遍历数组,如果该索引位的值大于target,说明返回值是当前i或者i-1,如果相等则为i,不相等则为i-1,要考虑数组头和数组尾的特殊情况。
public class Solution {
public int searchInsert(int[] nums, int target) {
for(int i=0;i<nums.length;i++) {
if(i==0 && nums[i]>=target)
return 0;
if (nums[i] > target) {
if (nums[i - 1] == target)
return i - 1;
else
return i;
}
if(i==nums.length-1){
if(nums[i] == target)
return i;
else
return nums.length;
}
}
return 0;
}
}
执行用时:0 ms, 在所有 Java 提交中击败了100.00% 的用户
内存消耗:37.9 MB, 在所有 Java 提交中击败了83.59% 的用户
2、二分查找
经典的二分查找思想,注意边界,后续可能会写一个专门注意二分查找边界的文章。
public class Solution {
public int searchInsert(int[] nums, int target) {
int left = 0;
int right = nums.length-1;
while(left<=right){
int mid = (left+right)/2;
if(nums[mid] == target)
return mid;
else if(nums[mid] < target)
left = mid+1;
else
right = mid -1;
}
return left;
}
}
执行用时:0 ms, 在所有 Java 提交中击败了100.00% 的用户
内存消耗:38 MB, 在所有 Java 提交中击败了75.60% 的用户