Leetcode 35. Search Insert Position

最新推荐文章于 2024-10-21 14:42:16 发布

转载最新推荐文章于 2024-10-21 14:42:16 发布 · 60 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/lettuan/p/6367232.html

本文介绍了一个算法问题的解决方案，即在一个已排序的数组中找到目标值的插入位置。该问题要求返回目标值若存在时的索引，或按顺序插入时的正确位置。通过二分搜索实现高效查找。

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

思路：通过观察可以发现，这一题其实就是要求array中>=target的第一个元素的位置。注意要将right设置成len(array)，因为如果array中没有>=target的数，最后的情况是left=right=len(array)。

 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         
 9         if not nums:
10             return 0
11             
12         n = len(nums)
13         left, right = 0, n
14         
15         while left < right:
16             mid = left + (right - left)/2
17             if nums[mid] >= target:
18                 right = mid
19             else:
20                 left = mid + 1
21         
22         return left
23