74. 搜索二维矩阵
题目
74. 搜索二维矩阵
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
解法
注意到输入的 m x n 矩阵可以视为长度为 m x n的有序数组。
该二维数组可以拉长为一个有序的一维数组。
由于该 虚 数组的序号可以由下式方便地转化为原矩阵中的行和列 (我们当然不会真的创建一个新数组) ,该有序数组非常适合二分查找。
row = idx // n , col = idx % n。
这是一个标准二分查找算法
初始化左右序号
- left = 0 和 right = m x n - 1。
While left < right : - 选取虚数组最中间的序号作为中间序号: pivot_idx = (left + right) / 2。
- 该序号对应于原矩阵中的 row = pivot_idx // n行 col = pivot_idx % n 列, 由此可以拿到中间元素pivot_element。该元素将虚数组分为两部分。
- 比较 pivot_element 与 target 以确定在哪一部分进行进一步查找。
public boolean searchMatrix(int[][] matrix, int target) {
int m = matrix.length;
int n = matrix[0].length;
int left = 0;
int right = m * n - 1;
int pivotIndex, pivotElement;
while (left <= right) {
// 二分并且计算出在二维矩阵中的值
pivotIndex = (left + right) / 2;
pivotElement = matrix[pivotIndex / n][pivotIndex % n];
// 判断中间值与目标值并且进行二分
if (pivotElement == target) {
return true;
} else if (pivotElement < target) {
left = pivotIndex + 1;
} else {
right = pivotIndex - 1;
}
}
return false;
}
剑指 Offer 04. 二维数组中的查找 & 240. 搜索二维矩阵 II
题目
剑指 Offer 04. 二维数组中的查找
240. 搜索二维矩阵 II
这两个题是一样的
在一个 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。
解法
二分查找
由于给定的二维数组具备每行从左到右递增以及每列从上到下递增的特点,当访问到一个元素时,可以排除数组中的部分元素。
从二维数组的右上角开始查找。如果当前元素等于目标值,则返回 true。如果当前元素大于目标值,则移到左边一列。如果当前元素小于目标值,则移到下边一行。
可以证明这种方法不会错过目标值。如果当前元素大于目标值,说明当前元素的下边的所有元素都一定大于目标值,因此往下查找不可能找到目标值,往左查找可能找到目标值。如果当前元素小于目标值,说明当前元素的左边的所有元素都一定小于目标值,因此往左查找不可能找到目标值,往下查找可能找到目标值。
- 若数组为空,返回 false
- 初始化行下标为 0,列下标为二维数组的列数减 1
- 重复下列步骤,直到行下标或列下标超出边界
-
- 获得当前下标位置的元素 num
-
- 如果 num 和 target 相等,返回 true
-
- 如果 num 大于 target,列下标减 1
-
- 如果 num 小于 target,行下标加 1
- 循环体执行完毕仍未找到元素等于 target ,说明不存在这样的元素,返回 false`
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return false;
}
int m = matrix.length;
int n = matrix[0].length;
// 从右上角开始遍历
int row = 0;
int column = n - 1;
// 一直遍历到左下角
while (row < m && column >= 0) {
int element = matrix[row][column];
if (element == target) {
return true;
} else if (element < target) {
row++;
} else {
column--;
}
}
return false;
}