剑指 0ffer 二维数组中的查找
2020-06-07 本文已影响0人
快乐的工程师
算法名称：二维数组中的查找
题目内容：在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
解题思路：
首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数字，剔除这个数字所在的列；如果该数字小于要查找的数字，剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。
例如，我们要在上述的二维数组中查找数字7的步骤如下图所示：
F3708B2CFD9F71157AA94175FAE464DB.jpg
实例代码：
- Java
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix.length \> 0 && matrix[0].length \> 0) {
int row = 0;
int colomn = matrix[0].length - 1;
while (row \< matrix.length && colomn \>= 0) {
if (target \< matrix[row][colomn]) {
colomn--;
} else if (target \> matrix[row][colomn]) {
row++;
} else if (target == matrix[row][colomn]) {
return true;
}
}
}
return false;
}
}
- Golang版
func findNumberIn2DArray(matrix [][]int, target int) bool {
if len(matrix) > 0 && len(matrix[0]) > 0 {
row := 0
colomn := len(matrix[0]) - 1
for row < len(matrix) && colomn >= 0 {
if target < matrix[row][colomn] {
colomn--
} else if target > matrix[row][colomn] {
row++
} else if target == matrix[row][colomn] {
return true
}
}
}
return false
}
本题考点：
- 考察面试者对二维数组的理解及编程能力。二维数组在内存中占据连续的空间。在内存中从上到下存储各行元素，在同一行中按照从左到右的顺序存储。因此我们可以根据行号和列号计算出相对于数组首地址的偏移量，从而找到对应的元素。
