LeetCode 35. 搜索插入位置
35. 搜索插入位置
Difficulty: 简单
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
Solution
Language: 全部题目
这是一道很简单很简单的数组题目,然而提交了三次才通过,只能说自己考虑问题太不全面了!
解法一(很脏):
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
if not nums: return 0
for i in range(len(nums)-1):
if target > nums[i] and target <= nums[i+1]:
return i+1
if target <= nums[i]:
return i
return len(nums) if target > nums[-1] else 0
解法二:
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
for i in range(len(nums)):
if nums[i] >= target:
return i
return len(nums)
下面是两次的错误提交记录:
错误解法一:如果测试用例为
[1]
0
测试不通过,数组的长度为零,此时for循环已经不起作用了,之间返回数组的长度,当target小于数组中唯一的元素时才能保持正确。
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
if not nums: return 0
for i in range(len(nums)-1):
if target > nums[i] and target <= nums[i+1]:
return i+1
if target < nums[i]:
return i
return len(nums)
错误解法二:由于粗心导致,for循环中除了target > nums[i]
还有target <= nums[i]
而不是target < nums[i]
,所以导致
[1,3,5]
1
测试用例无法通过。
class Solution:
def searchInsert(self, nums: List[int], target: int) -> int:
if not nums: return 0
for i in range(len(nums)-1):
if target > nums[i] and target <= nums[i+1]:
return i+1
if target < nums[i]:
return i
return len(nums) if target > nums[-1] else 0