leetcode 日经贴，Cpp code -Search a 2D Matrix II

原创于 2015-07-23 11:36:52 发布 · 299 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

173 篇文章

订阅专栏

Cpp

152 篇文章

订阅专栏

本文介绍了一个高效的二维矩阵搜索算法，该算法能在O(n)的时间复杂度内查找目标元素是否存在。通过遍历矩阵右上角开始，根据目标值与当前值大小比较决定向左还是向下移动，直至找到目标值或遍历结束。

Search a 2D Matrix II

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