Leetcode：240. 搜索二维矩阵 II

云格～2024-06-03 17:18

题目

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

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

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

示例 1：

示例 2：

提示：

m == matrix.length

n == matrix[i].length

1 <= n, m <= 300

-109 <= matrix[i][j] <= 109

每行的所有元素从左到右升序排列

每列的所有元素从上到下升序排列

-109 <= target <= 109

思路

我们将矩阵逆时针旋转 45° ，并将其转化为图形式，发现其类似于二叉搜索树

代码

cpp 复制代码

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int i = matrix.size()-1,j = 0;
        while(i>=0&&j < matrix[0].size()){
            if(target > matrix[i][j]) j++;
            else if(target < matrix[i][j]) i--;
            else return true;
        }
        return false;
    }
};