给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
第一次
首先判断target在不在nums里面,在,输出位置;不在,另一种做法
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
n = len(nums)
if target in nums:
for i in range(n):
if nums[i] == target:
return i
else:
if target < nums[0]:
return 0
elif target > nums[-1]:
return n
for i in range(n - 1):
if nums[i] < target < nums[i + 1]:
return i + 1
第二次
第一次时,还需要判断该数字在不在里面,能否省略判断,即边检测边判断,于是出现第二版本
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
n = len(nums)
if target < nums[0]:
return 0
if target > nums[-1]:
return n
if target == nums[-1]:
return n - 1
for i in range(n - 1):
if target == nums[i]:
return i
elif nums[i] < target < nums[i + 1]:
return i + 1
第三次
第三次更改,嗯,感觉已经很完美了
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
n = len(nums)
if target > nums[-1]:
return n
for i in range(n):
if target == nums[i]:
return i
elif target < nums[i]:
return i
第四次
事实证明,我就是个弟弟。这里考的是二分算法,等到有时间时补充。
too young too simple, study hard!