给定M×N矩阵,每一行、每一列都按升序排列,请编写代码找出某元素。
示例:
现有矩阵 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。
分析:
方法:DFS
如果从第一行第一列的数开始遍历,它的右方和下方都有比它大的数,找不到方向。如果从第一行最后一列的开始遍历,左边的数一定比当前数小,下方的数一定比当前数大,因此只需要判断目标数与当前数的大小,就可以找到遍历的方向。我们可以通过深度遍历(DFS)的方式实现,递归参数就为横纵坐标,递归终止条件就是越界返回 false,找到目标数返回 true。
时间复杂度:O(m+n) m、n 为矩阵宽高,最坏情况目标数在对角线尽头。
空间复杂度:O(m+n)
class Solution {
int[][] matrix;
int target;
public boolean searchMatrix(int[][] matrix, int target) {
//数组为空
if(matrix.length == 0){
return false;
}
this.matrix = matrix;
this.target = target;
return dfs(0, matrix[0].length-1);
}
public boolean dfs(int i, int j){
//越界
if(i >= matrix.length || j < 0){
return false;
}
//找到目标数
if(matrix[i][j] == target){
return true;
}
//小于目标数,往下
if(matrix[i][j] < target){
return dfs(i+1, j);
}
//大于目标数,往左
return dfs(i, j-1);
}
}
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sorted-matrix-search-lcci