LeetCode 240 搜索二维矩阵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

提示

提示：

  • m == matrix.length
  • n == matrix[i].length
  • 1 <= n, m <= 300
  • $-10^9 \leq$ matrix[i][j] $\leq 10^9$
  • 每行的所有元素从左到右升序排列
  • 每列的所有元素从上到下升序排列
  • $-10^9 \leq$ target $\leq 10^9$

AC代码

从右上角建根，可以看成抽象的二叉树。

1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m = matrix.size(), n = matrix[0].size();
        int l = 0, r = n - 1;
        while(l < m && r >= 0){
            if(matrix[l][r] == target) return true;
            if(matrix[l][r] > target) r--;
            else l++;
        }
        return false;
    }
};