LeetCode:21.搜索二维矩阵Ⅱ

搜索二维矩阵Ⅱ

目录

1.暴力枚举

2.Z字型查找

---

1.暴力枚举

对于在一个二维矩阵中查找元素是否存在，我们首先会想到暴力枚举，通过两次遍历可以找出

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

2.Z字型查找

对于第一种方法我们是否可以进行优化，这个二维矩阵每行从左到右，每列从上到下升序，可以利用这一特性，从右上角第一个元素开始判断，如果大于目标值，因为这个元素是这一行的最大元素了，应当去前面继续查找，同理，小于目标值，去下一行查找

通过这样判断，我们可以一次排除掉一行或一列的元素，相比于两次遍历，时间复杂度有所提升

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