leetcode hot100 之 搜索二维矩阵II

题目

给定一个矩阵和一个目标值，请判断目标值是否在矩阵当中。该矩阵有如下特征：
每列从上往下升序排列；
每行从左往右升序排列。

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

原题链接：https://leetcode.cn/problems/search-a-2d-matrix-ii/

思路

从左下角开始出发，如果目标值比当前值小，则往上走；如果目标值比当前值大，则往右走。找到返回即可。

• 复杂度分析
  • 时间复杂度 O(m+n)。m、n 分别为矩阵的行数、列数。
  • 空间复杂度 O(1)。

代码

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