一、题目描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
二、解决办法
1. 解决办法一:
采用暴力求解的方式,算法复杂度O(N) (Python实现)。
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
'''
# 暴力求解,复杂度O(n)
n = len(nums)
for i in range(n):
if nums[i] >= target:
return i
return n
'''
2. 解决办法二:
采用二分法进行求解,算法复杂度Olog(N)。其中二分法根据开闭区间,提供两种思路:
思路一:左右两侧都是闭区间 (Python实现)。
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
'''
#二分查找
n = len(nums)
left = 0
right = n-1
while (left <= right):
mid = int((left + right)/2)
if nums[mid] > target:
right = mid-1
elif nums[mid] < target:
left = mid+1
else :
return mid
return right + 1
思路二:左侧闭区间,右侧开区间 (C++实现)。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int n = nums.size();
int left = 0;
int right = n - 1; // 定义target在左闭右闭的区间里,[left, right]
while (left <= right) { // 当left==right,区间[left, right]依然有效
int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
if (nums[middle] > target) {
right = middle - 1; // target 在左区间,所以[left, middle - 1]
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,所以[middle + 1, right]
} else { // nums[middle] == target
return middle;
}
}
// 分别处理如下四种情况
// 目标值在数组所有元素之前 [0, -1]
// 目标值等于数组中某一个元素 return middle;
// 目标值插入数组中的位置 [left, right],return right + 1
// 目标值在数组所有元素之后的情况 [left, right], return right + 1
return right + 1;
}
};
三、算法比较
下图的第一个通过是二分法求解,第二个通过是暴力求解。可以看出在这道题目中,二分法查找和暴力求解的运行时间差不多,二分法的内存消耗比暴力的循环少一点。
1. 暴力求解的具体时间和空间复杂度;
2. 二分法的具体时间和空间复杂度:
参考文献:
1. https://mp.weixin.qq.com/s/fCf5QbPDtE6SSlZ1yh_q8Q
2. https://www.bilibili.com/video/BV187411d77t?from=search&seid=4957648913905534017