题目描述:给定一个矩阵和target,判断target是否在矩阵中。矩阵每一行是单调的,每一列也是单调递增的。

题目链接:Leetcode 240. Search a 2D Matrix II

这题不可以像74题一样直接判断二分了,这个并不可以展开成一段段区间的二分查找,这个需要每一行每一行进行二分查找。二分查找的递归版本和迭代版本各不相同。

但是仔细看了看其他人的答案,是从左上角搜索起来,这个真的难以想到。

class Solution:
    def searchMatrix(self, matrix, target):
        if not matrix:
            return False
        m, r, c = len(matrix), 0, len(matrix[0]) - 1
        while r < m and c >= 0:
            if matrix[r][c] == target:
                return True
            elif matrix[r][c] > target:
                c -= 1
            elif matrix[r][c] < target:
                r += 1      
        return False

代码如下

class Solution:
    def searchMatrix(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        if not matrix or not matrix[0]:
            return False
        m,n = len(matrix),len(matrix[0])
        for r in matrix:
            if self.binarySearch(r,target):
                return True
        return False
        
        
    def binarySearch(self,row,target):
        if not row:
            return False
        mid = len(row) // 2
        if row[mid] == target:
            return True
        if row[mid] > target:
            return self.binarySearch(row[:mid],target)
        if row[mid] < target:
            return self.binarySearch(row[mid+1:],target)
        

参考链接

Leetcode 240. Search a 2D Matrix II

相关文章: