引言
力扣题目【搜索二维矩阵】描述 -> 传送门
所用语言:Java
执行用时 0 ms,内存消耗 37.5 MB
正文
今天的题目,虽然官方显示是中等难度,但是我个人觉得这个应该算是简单题,因为题目给的条件太宽松了:①矩阵是有序排列的;②矩阵的长、宽很小,让人没有想要去优化的欲望 😶
先说说我的思路吧:
- 根据题目描述,矩阵的行数据是升序排列的,矩阵的列数据也是升序排列的
- 所以,可以先锁定
target
可能出现在哪一行,然后再去看这一行中是否有target
我下面给出的答案就是按照这种思路来的,请看:
答案
public boolean searchMatrix(int[][] matrix, int target) {
// 矩阵为null,直接返回
if (matrix == null) {
return false;
}
int row = 0;
// 先确定target可能会出现在哪一行
while (row + 1 < matrix.length) {
if (matrix[row][0] <= target && target < matrix[row + 1][0]) {
break;
} else {
row++;
}
}
// 进一步判断target是否存在
for (int i = 0; i < matrix[row].length; i++) {
if (matrix[row][i] == target) {
return true;
}
}
return false;
}
诚如上面答案所示,我并没有刻意用到什么查找算法,因为矩阵的最大行宽和列宽是 100,仅是简单的遍历,就能达到使执行时间是 0 ms的效果了。在上面的代码中,最坏的情况也不过是比较 200 次。
当然,如果你想要优化,可以在行搜索和列搜索中使用二分查找,因为它们的数据都是有序的。
最后
祝大家多多AC 😊