题目描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
例如:输入: [1,3,5,6], 5 输出: 2 ;输入: [1,3,5,6], 7 输出: 4 ;输入: [1,3,5,6], 0 输出: 0。
解题思路:用暴力解决的话,可以对数组进行遍历,然后逐个与目标值进行比对,时间复杂度是O(n)。还有一种方法就是二分查找,二分查找时间复杂度为O(logn) 。
先定义好左右下标left和right,并用(left + right) / 2算出中间下标mid,然后每次循环先判断中间下标的值与目标值是否相等,相等则返回中间下标mid。
int mid = (left + right) / 2;
if(nums[mid] == target) {
return mid;
如果nums[mid] < 目标值,则说明目标值在数组的右半部位,将left右移到mid的下一位的位置,left = mid + 1; 如果nums[mid] > 目标值,则说明目标值在数组的左半部位,将right左移到mid的前一位的位置,right = mid - 1。然后继续计算新的mid的值,再判断中间下标的值与目标值是否相等,再继续循环,直到left大于right时跳出循环。如果循环结束还没有在数组中找到与目标值相等的值,则返回left,即其将会被插入的索引值。
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;
}
}
总的解题代码如下:
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;
}
}