240. 搜索二维矩阵 II-CSDN博客

本文链接：https://blog.csdn.net/qq_38023194/article/details/140985632

240. 搜索二维矩阵 II

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性：

每行的元素从左到右升序排列。
每列的元素从上到下升序排列。

示例 1：
在这里插入图片描述

输入：matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 5
输出：true

示例 2：
在这里插入图片描述

输入：matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 20
输出：false

解题思路

由于矩阵 matrix 中每一行的元素都是升序排列的，因此我们可以对每一行都使用一次二分查找，判断 target 是否在该行中，从而判断 target 是否出现。

def binary_search(nums, target):
	left, right = 0, len(nums)-1
	while left <= right:
		mid = (left + right) // 2
		if nums[mid] == target:
			return mid
		elif nums[mid] < target:
			left = mid + 1
		else:
			right = mid - 1
	else:
		return None

def searchMatrix(matrix, target):
	if not matrix:
		return False
	for row in matrix:
		index = binary_search(row, target)
		if index is not None:
			return True
	else:
		return False