目录

0074:搜索二维矩阵(★)

力扣第 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

相似问题:

分析

先二分查找定位是哪一行,再在该行中二分查找即可。

解答

1
2
3
4
5
class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        i = bisect_right(matrix,target,key=lambda x:x[0])-1
        j = bisect_right(matrix[max(0,i)],target)-1
        return matrix[max(0,i)][max(0,j)]==target

时间复杂度 O(log(M*N)),40 ms