刷题-力扣-面试题 10.09. 排序矩阵查找

题目链接

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sorted-matrix-search-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目描述

给定M×N矩阵,每一行、每一列都按升序排列,请编写代码找出某元素。

示例:

现有矩阵 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。

题目分析

  1. 根据题目描述,矩阵按行列递增,从矩阵得左下角开始,向右一次增大、向上一次减小
  2. 二分查找,从矩阵左下角开始查找

代码

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        if (matrix.size() == 0) { return false; }

        int r = matrix.size() - 1;
        int c = 0;
        while (r >= 0 && c < matrix[0].size()) {
            if (matrix[r][c] == target) { return true; }
            else if (matrix[r][c] > target) { --r; }
            else { ++c; }
        }
        return false;
    }
};
posted @ 2022-03-09 13:35  韩亚光  阅读(33)  评论(0编辑  收藏  举报