74.搜索二维矩阵
题目:
给你一个满足下述两条属性的 m x n
整数矩阵:
- 每行中的整数从左到右按非严格递增顺序排列。
- 每行的第一个整数大于前一行的最后一个整数。
给你一个整数 target
,如果 target
在矩阵中,返回 true
;否则,返回 false
。
示例 1:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3 输出:true
示例 2:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13 输出:false
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 100
-104 <= matrix[i][j], target <= 104
解题思路:
本题的主要想法是两次二分查找,首先查找每一行的第一个值和target的关系,找到第一个比target大或者遍历完所有的列,然后想回找最后一个比target小的行,利用二分查找当前的行
代码:
class Solution:def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:m = len(matrix)n = len(matrix[0])i = 0while i<m and target>=matrix[i][0]:i+=1left, right = 0, n-1while left<=right:mid = (left+right)//2if target==matrix[i-1][mid]:return Trueelif target>matrix[i-1][mid]:left = mid+1else:right =mid-1return False