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

一、题目

题目链接：力扣

在一个 n * m 的二维数组中，每一行都按照从左到右 非递减 的顺序排序，每一列都按照从上到下 非递减 的顺序排序。请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

示例:

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

限制：

0 <= n <= 1000

0 <= m <= 1000

---

二、题解

1、思路

🐶 最初的灵感，但貌似没用，直接忽略

🐶 暴力破解：

下面这种情况在数组为空的时候会出错，因为数组为空n为0，第二行又去访问第0行，是不存在的，因此会报错。

int n = matrix.size();

int m = matrix[0].size();// 容易出错

if(m == 0 && n == 0)return false;

🐶 二分查找

函数lower_bound()在first和last中的前闭后开区间进行二分查找，返回大于或等于val的第一个元素位置。如果所有元素都小于val，则返回last的位置（注意判断）。

🐶 Z字形查找法

数组左下角向右上角看，使用副对角线，而不是主对角线：

 从左下角3开始，画一个V字形，左上比3小，右上角比3大。如果target为8，则有：

从矩阵 matrix 左下角元素（索引设为 (i, j) ）开始遍历：

当 matrix[i][j] > target 时，执行 i-- ，即消去第 i 行元素；
当 matrix[i][j] < target 时，执行 j++ ，即消去第 j 列元素；
当 matrix[i][j] = target 时，返回 truetrue ，代表找到目标值。

2、代码实现

🐶 暴力破解：

class Solution {
public:
    bool findNumberIn2DArray(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;
        for(int i = 0; i < n; i++)
        {
            for(int j = 0; j < m; j++)
            {
                if(matrix[i][j] == target)return true;
            }
        }

        return false;
    }
};

🐶 二分查找

class Solution {
public:
    bool findNumberIn2DArray(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 count = 0;
        for(int i = 0; i < n; i++)
        {
            auto it = lower_bound(matrix[i].begin(), matrix[i].end(), target);
            // 这里有个坑，需要首先判断it是不是end，不能直接取值，否则可能出错
            if(it != matrix[i].end() && *it == target)return true;
        }

        return false;
    }
};

🐶 Z字形查找法

class Solution {
public:
    bool findNumberIn2DArray(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;
        
        // 一定要搞清楚m和n，别用错
        // 注意“i >= 0 &&”，一定要是大于等于才行
        for(int i = n - 1, j = 0; i >= 0 && j < m;)
        {
            if(matrix[i][j] > target)i--;
            else if(matrix[i][j] < target)j++;
            else if(matrix[i][j] == target)return true;
        }
        return false;
    }
};

3、复杂度分析

🐶 暴力破解：

时间复杂度：O(nm)；

空间复杂度：O(1)。

🐶 二分查找

时间复杂度：O(nlogm)；

空间复杂度：O(1)。

🐶 Z字形查找法

时间复杂度：O(n+m)；// 最好的情况下是弯折的对角线（可能m!=n），最坏是沿着两边。

空间复杂度：O(1)。

4、运行结果

🐶 暴力破解：

 🐶 二分查找

🐶 Z字形查找法

---

三、参考内容

力扣

力扣