码迷,mamicode.com
首页 > 其他好文 > 详细

leedcode35之搜索插入位置

时间:2020-06-22 01:14:17      阅读:44      评论:0      收藏:0      [点我收藏+]

标签:search   相等   print   col   位置   重要   ==   get   搜索   

题目描述

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例 1:

输入: [1,3,5,6], 5
输出: 2

代码:
 1 def binarySearch(nums, target):
 2     ‘‘‘
 3 
 4     :param nums:
 5     :param target:
 6     :return:
 7     ‘‘‘
 8     left, right = 0, len(nums) - 1
 9     while left <= right:
10         # (right+left)//2存在内存溢出的情况,故采用下面更优!
11         mid = left + (right - left) // 2
12         if nums[mid] == target:
13             return mid
14         elif nums[mid] < target:
15             left = mid + 1
16         else:
17             right = mid - 1
18 
19     return left
20 
21 
22 print("----------测试binarySearch()---------")
23 nums = [1, 2, 4, 6, 8, 9]
24 index = binarySearch(nums, 2)
25 print("index=", index)
26 
27 
28 def binarySearch1(nums, target):
29     ‘‘‘
30 
31     :param nums:
32     :param target:
33     :return:
34     ‘‘‘
35     left, right = 0, len(nums) - 1
36     while left < right:
37         mid = left + (right - left) // 2
38         if nums[mid] == target:
39             return mid
40         elif nums[mid] < target:
41             left = mid + 1
42         else:
43             right = mid
44 
45     # 此时left==right
46     if target <= nums[left]:
47         return left
48     else:
49         return left + 1
50 
51 
52 print("----------测试binarySearch1()---------")
53 nums = [1, 3, 5, 6]
54 index = binarySearch1(nums, 7)
55 print("index=", index)

输出:

----------测试binarySearch()---------
index= 1
----------测试binarySearch1()---------
index= 4

总结:在二分查找中,最重要的是边界条件判断。在方法二中,判断条件为Left<right;当跳出循环时,left=right,此时元素nums[left]并未加入判断。所以需要在后面再进一步判断targte是否和nums[left]相等,方法中采用的<=,满足则返回left,反之返回left+1

leedcode35之搜索插入位置

标签:search   相等   print   col   位置   重要   ==   get   搜索   

原文地址:https://www.cnblogs.com/rounie/p/13174548.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!