leetcode 240. 搜索二维矩阵 II

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target。该矩阵具有以下特性：

每行的元素从左到右升序排列。
每列的元素从上到下升序排列。

---

示例:

现有矩阵 matrix 如下：

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]
给定 target = 5，返回 true。

给定 target = 20，返回 false。

---

二分查找

class Solution(object):
    def searchMatrix(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        def findElement(nums, target):
            if not nums:
                return False
            right = len(nums) - 1 
            left = 0
            mid = right >> 1
            if nums[mid] == target:
                return True
            elif nums[mid] < target:
                return findElement(nums[mid+1:], target)
            elif nums[mid] > target:
                return findElement(nums[:mid], target)
                
        if len(matrix) == 0 or len(matrix[0]) == 0:
            return False
        rows = len(matrix)
        cols = len(matrix[0])
        for i in range(rows):
            if matrix[i][0] <= target and matrix[i][-1] >= target:
                if findElement(matrix[i], target):
                    return True
        return False
    

由于都是已经排好序的所以可以使用二分查找。对矩阵进行逐行的检查，首先检查每行的首尾，判断target是否在其中，如果在那么对行进行二分查找，如果不在那么继续找下一行。如果最后都没找到，返回False。

迭代

class Solution(object):
    def searchMatrix(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        def findElement(nums, target):
            if not nums:
                return False
            right = len(nums) - 1 
            left = 0
            mid = right >> 1
            if nums[mid] == target:
                return True
            elif nums[mid] < target:
                return findElement(nums[mid+1:], target)
            elif nums[mid] > target:
                return findElement(nums[:mid], target)
                
        if len(matrix) == 0 or len(matrix[0]) == 0:
            return False
        rows = len(matrix)
        cols = len(matrix[0])
        for i in range(rows):
            if matrix[i][0] <= target and matrix[i][-1] >= target:
                if findElement(matrix[i], target):
                    return True
        return False
    

递归的二分查找，空间复杂度为O(1)，比递归的方法更节省空间。