搜索二维矩阵
给你一个满足下述两条属性的 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
这个是简单的二维矩阵搜索,因为这个可以看成是把一个列表折成了几段,然后这样放在一起了,所以直接二分查找就好了
只不过是要用变量变通一下即可,核心还是二分查找的代码
class Solution:
def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
left = 0
right = len(matrix)*len(matrix[0])-1
while left <= right :
mid = (left+right)//2
r,c = mid//len(matrix[0]),mid%len(matrix[0])
if matrix[r][c] == target:return True
if matrix[r][c]<target:left = mid+1
else:right = mid-1
return False