算法题-搜索二维矩阵 II

题目描述：

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

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

示例 1：

输入：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

示例 2：

输入：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 = 20
输出：false

思路：

参考灵神的思路：

从右上角开始，寻找6

5 < 6，意味着第一排的所有数都小于6，排除

剩余矩阵中比较右上角元素和6的大小。

10 > 6，这意味着最后一列的左右数都大于6，排除

在剩余矩阵中，比较右上角元素和6的大小。

9 > 6，意味着最后一列的所有数都大于6，排除。

右上角元素等于6，找到。

自己写的：

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        # 遍历每一行最后一个元素i 如果i小于target 则这一行都可以去掉
        # 然后遍历下一行最后一个元素i，如果i大于target 则这一列都去掉
        n = len(matrix[0])
        for i in range(len(matrix)):
            value = matrix[i][n-1]
            print("value:", value)

            if value < target:
                continue
            elif value > target:
                n -= 1
                if n == 0:
                    return False
                for j in range(n):
                    if matrix[i][j] == target:
                        return True
            elif value == target:
                return True
            
        

参考灵神：

python：

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        # 遍历每一行最后一个元素i 如果i小于target 则这一行都可以去掉
        # 然后遍历下一行最后一个元素i，如果i大于target 则这一列都去掉
        m, n = len(matrix), len(matrix[0])
        i, j = 0, n - 1 # 从右上角开始
        while i < m and j >= 0: # 还有剩余元素
            if matrix[i][j] == target:
                return True # 找到target
            if matrix[i][j] < target:
                i += 1 # 这一行剩余元素都小于target，排除
            else:
                j -= 1 # 这一列剩余元素都大于target，排除
        return False
            
        

java：

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int i = 0;
        int j = matrix[0].length - 1; // 从右上角开始
        while (i < matrix.length && j >= 0) { // 还有剩余元素
            if (matrix[i][j] == target) {
                return true; // 找到target
            }
            if (matrix[i][j] < target) {
                i++; // 这一行剩余元素全部小于target，排除
            } else {
                j --; // 这一列剩余元素全部大于target，排除
            }
        }
        return false;
    }
}