剑指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. 二维数组中的查找 作者源代码 - 牛客网