LeetCode-240-搜索二维矩阵 II

1、二分查找

我们可以对每一行进行二分查找，若能查找到则说明存在。时间复杂度为O(mlogn)。

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        for (const auto& row: matrix) {
            auto it = lower_bound(row.begin(), row.end(), target);
            if (it != row.end() && *it == target) {
                return true;
            }
        }
        return false;
    }
};

2、副对角线查找

我们以[0,matrix[0].size() - 1]作为查找区域的右上角，以原矩阵的左下角作为查找区域的左下角。当右上角元素的坐标仍然在矩阵内时：1、若matrix[row][col]=target，说明此时我们已经查找到元素，返回true；2、若matrix[row][col]<target，说明此时右上角元素小于目标元素，我们只能在当前行以下的部分找到目标元素，故++row；3、若matrix[row][col]>target，说明此时右上角元素大于目标元素，我们只能在当前列以左的部分找到目标元素，故–col。时间复杂度为O(m+n)。

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