剑指 Offer 04. 二维数组中的查找

思路

(1) 暴力法：遍历整个二维数组，时间复杂度为O(n*m)

(2) 二分查找：对每一行进行二分查找，时间复杂度为O(n*logm)，但这样没有用到"每一列都按照从上到下递增的顺序排序"这个条件

(3) 将矩阵旋转45度，讲解如下：

 

代码实现

class Solution {
public:
    bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
        //n行m列
        int n = matrix.size();
        if(n == 0)
            return false;
        int m = matrix[0].size();
        if(m == 0)
            return false;

        //从右上角元素开始搜索(左下角开始也行)
        int i = 0, j = m-1;
        while(1) {
            if(matrix[i][j] == target) {
                return true;
            } else if (matrix[i][j] < target) {
                ++i; //消去第i行
                if(i >= n) 
                    break;
            } else {
                --j; //消去第j列
                if(j < 0) 
                    break;
            }
        }
        return false;
    }
};

复杂度分析

参考

面试题04. 二维数组中的查找（标志数，清晰图解）