搜索二维矩阵
- leetcode 74. 搜索二维矩阵
- 题目描述
- 抽象BST
- 代码演示
- 抽象BST
leetcode 74. 搜索二维矩阵
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/search-a-2d-matrix
题目描述
给你一个满足下述两条属性的 m x n 整数矩阵:
每行中的整数从左到右按非递减顺序排列。
每行的第一个整数大于前一行的最后一个整数。
给你一个整数 target ,如果 target 在矩阵中,返回 true ;否则,返回 false 。
示例1:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
输出:true
示例2:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
输出:false
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 100
-10000 <= matrix[i][j], target <= 10000
抽象BST
我们可以将二维矩阵抽象成「以右上角为根的 BST」:
那么我们可以从根(右上角)开始搜索,如果当前的节点不等于目标值,可以按照树的搜索顺序进行:
当前节点「大于」目标值,搜索当前节点的「左子树」,也就是当前矩阵位置的「左方格子」,即 y—
当前节点「小于」目标值,搜索当前节点的「右子树」,也就是当前矩阵位置的「下方格子」,即 x++
代码演示
public boolean searchMatrix(int[][] matrix, int target) {
if (matrix.length == 0 || matrix[0].length == 0){
return false;
}
int n = matrix.length;
int m = matrix[0].length;
int c = m - 1;
int r = 0;
while (r < n && c >= 0){
if (matrix[r][c] < target){
r++;
} else if (matrix[r][c] > target) {
c--;
}else{
return true;
}
}
return false;
}
抽象BST
剑指 Offer 04. 二维数组中的查找