首页 > 代码库 > 35. Search Insert Position

35. Search Insert Position

2024-09-08 11:21:19   208人阅读

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.

Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0

 

自己写的~~

 1 class Solution(object):
 2     def searchInsert(self, nums, target):
 3         """
 4         :type nums: List[int]
 5         :type target: int
 6         :rtype: int
 7         """
 8         if target < nums[0]:
 9             return 0
10         for i in nums:
11             if i >= target:
12                 return nums.index(i)
13         return len(nums)

注意讨论边界情况啊！

 

1 class Solution(object):
2     def searchInsert(self, nums, target):
3         """
4         :type nums: List[int]
5         :type target: int
6         :rtype: int
7         """
8         return len([x for x in nums if x<target])
[x for x in nums if x<target]生成一个比target小的列表

 1 class Solution(object):
 2     def searchInsert(self, nums, target):
 3         """
 4         :type nums: List[int]
 5         :type target: int
 6         :rtype: int
 7         """
 8         ans = 0
 9         low,high,mid = 0,len(nums)-1,0
10         while(low <= high):
11             mid = (low +  high)/2
12             if nums[mid] > target:
13                 high = mid - 1
14             elif nums[mid] < target:
15                 low = mid + 1
16             else:
17                 return mid
18         if high < mid:
19             return mid
20         elif low > mid:
21             return low

二分法~

35. Search Insert Position


联系
我们