原题链接: https://leetcode.com/problems/search-insert-position/
1. 题目介绍
Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.
You may assume no duplicates in the array.
给出一个有序数组和一个目标值,如果目标值在数组中,那么就返回目标值的下标。如果目标值不在数组中,那么就返回它应该排在这个有序数组中的下标。
你可以假设,这个数组中没有重复的元素。
Example 1:
Input: [1,3,5,6], 5
Output: 2
Example 2:
Input: [1,3,5,6], 2
Output: 1
Example 3:
Input: [1,3,5,6], 7
Output: 4
Example 4:
Input: [1,3,5,6], 0
Output: 0
2. 解题思路
这道题一个典型的思路就是二分搜索。将数组从中间分成两个部分,判断 target 在左半部分还是右半部分,调用递归函数一直二分搜索下去。
时间复杂度 O( logN )
实现代码
class Solution {
public int searchInsert(int[] nums, int target) {
return helper( 0 , nums.length - 1 , nums ,target);
}
public int helper(int l, int r, int [] nums,int target){
if (l == r) {
if (target > nums[l]) {
return l+1;
}else{
return l;
}
}
int mid = ( l + r ) / 2;
if (target <= nums[mid]) {
return helper( l, mid , nums , target);
}else{
return helper( mid+1 , r , nums , target);
}
}
}
当然,二分搜索也可以不用递归,直接使用循环。代码会简洁一些,但是比较难懂。
实现代码
class Solution {
public int searchInsert(int[] nums, int target) {
int l = 0, r= nums.length-1;
while(l <= r ){
int mid = (l + r) / 2 ;
if(nums[mid] == target){
return mid;
}else if(nums[mid] > target){
r = mid-1;
}else{
l = mid+1;
}
}
return l;
}
}