python leetcode 74. Search a 2D Matrix 240. Search a 2D Matrix II

以此为基础的进阶题还有378. Kth Smallest Element in a Sorted Matrix

Search a 2D Matrix

class Solution(object):
    def searchMatrix(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        m=len(matrix)
        if m==0:
            return False
        n=len(matrix[0])
        i=0
        j=n-1
        while i<m and j>=0:
            if matrix[i][j]==target:
                return True 
            elif matrix[i][j]<target:
                i+=1
            else:
                j-=1 
        return False

Search a 2D Matrix II

class Solution(object):
    def searchMatrix(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        m = len(matrix)
        if  m==0:
            return False
        n = len(matrix[0])
        
        i=0
        j=n-1
        while i<m and j>=0:
            if target == matrix[i][j]:
                return True 
            elif target > matrix[i][j]:
                i+=1 
            else:
                j-=1
        return False