目录
二分查找
描述
请实现无重复数字的升序数组的二分查找
给定一个 元素升序的、无重复数字的整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标(下标从 0 开始),否则返回 -1
示例1
输入:[-1,0,3,4,6,10,13,14],13
返回值:6
示例2
输入:[],3
返回值:-1
代码实现
class Solution:
def binary_search(self , nums: List[int], target: int) -> int:
# write code here
left = 0
right = len(nums) -1
while left <= right:
cur = (left+right)//2
if nums[cur] > target:
right = cur -1
elif nums[cur] < target:
left = cur + 1
else:
return cur
return -1
二维数组中的查找
描述
在一个二维数组array中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
[[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]]
给定 target = 7,返回 true。
给定 target = 3,返回 false。
思路
由于给定的二维数组具备每行从左到右递增以及每列从上到下递增的特点,当访问到一个元素时,可以排除数组中的部分元素。
从二维数组的右上角开始查找。如果当前元素等于目标值,则返回 true。如果当前元素大于目标值,则移到左边一列。如果当前元素小于目标值,则移到下边一行。
可以证明这种方法不会错过目标值。如果当前元素大于目标值,说明当前元素的下边的所有元素都一定大于目标值,因此往下查找不可能找到目标值,往左查找可能找到目标值。如果当前元素小于目标值,说明当前元素的左边的所有元素都一定小于目标值,因此往左查找不可能找到目标值,往下查找可能找到目标值。
- 若数组为空,返回 false
- 初始化行下标为 0,列下标为二维数组的列数减 1
- 重复下列步骤,直到行下标或列下标超出边界
- 获得当前下标位置的元素 num
- 如果 num 和 target 相等,返回 true
- 如果 num 大于 target,列下标减 1
- 如果 num 小于 target,行下标加 1
- 循环体执行完毕仍未找到元素等于 target ,说明不存在这样的元素,返回 false
代码实现
def binary_search_2(target,arr):
rows = len(arr)
cols = len(arr[0])
row = 0
col = cols-1
while (row < rows and col >= 0 ):
cur = arr[row][col]
if cur == target:
return True
elif cur > target:
col -= 1
else:
row += 1
return False