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

发表于 | 更新于 | 分类于 | 阅读次数：

二维数组中的查找

题目：

在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

解法1：利用矩阵的特点

把二维数组看作一个矩阵，选择矩阵右上角或者左下角的元素开始比较。逐步缩小比较范围。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
public class Solution {
    public boolean Find(int target, int[][] matrix) {
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
            return false;

        int row = 0, column = matrix[0].length - 1; // 从右上角开始
        while (row <= matrix.length - 1 && column >= 0) {
            if (target == matrix[row][column])
                return true;
            else if (target > matrix[row][column])
                row++;
            else
                column--;
        }
        return false;
    }
}

解法2：

从每一行开始，对每一行进行二分查找。

解法3：时间复杂度为$O(n^2)$

从左到右从上到下遍历数组，这是最慢的方法。

1
2
3
4
5
6
7
8
9
10
11
12
public class Solution {
    public boolean Find(int target, int [][] array) {
        for(int i = 0; i < array.length; i++){
            for(int j = 0; j < array[0].length; j++){
                if(target == array[i][j]){
                    return true;
                }
            }
        }
        return false;
    }
}

拓展

如果此题修改为【判断该数组中是否存在该整数，如果存在，统计该整数出现的次数。】该怎么做？

相关链接

  1. 二维数组中的查找 作者源代码 - Github
  2. 二维数组中的查找 作者源代码 - 牛客网