刷题记录——剑指 Offer 04. 二维数组中的查找

2021/9/24 刷题记录

题目描述

在一个 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

我原本的思路

从第一行第一列开始查找，若target>matrix[i][j]则可以向下或向右，若target<matrix[i][j]则可以向上或向左。
该思路的漏洞：
1.满足一个条件时有两种操作，且未对两种操作进行具体的区分描述；
2.由于没有区分两种操作，会陷入死循环。举个例子：就按题目给定的矩阵，查找元素5。
（1）matrix[i][j] = 1, 5 > 1, matrix[i][j]更新为2。（假设先执行向下的操作）
（2）matrix[i][j] = 2, 5 > 2, matrix[i][j]更新为3.
（3）matrix[i][j] = 3, 5 > 3, matrix[i][j]更新为10.
（4）matrix[i][j] = 10, 5 < 10, matrix[i][j]更新为3.（假设先执行向上的操作）
由此就陷入了死循环。

题解思路

（加入了我的理解）
按照数组中元素排列的规律：右边和下边的数都比它大，左边和上边的数都比它小。那么以某元素为中心，该数组就可以划分为两部分，如图：

从左下角元素出发，要么向上走，要么向右走。这样就只有两种操作，且两种操作执行的条件不同。这样就可以不重复的遍历数组。

代码

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