力扣原题地址:
今天刷寻找目标值(二维数组中的查找),大家有兴趣可以点上看看题目要求,试着做一下。
我们直接看题解吧:
方法2,利用二叉搜索树
审题目+事例+提示:
题目意思是这个二维数组
每一行都按照从左到右递增的顺序排序,
每一列都按照从上到下递增的顺序排序
思路(二叉搜索树):
根节点对应的元素是左下角与右上角的两个元素,
若以左下角为例:
1、从左下角元素二维数组plants[i][j]开始遍历,比较对应target元素
·当plants[i][j]>target ,则i--,即消去i元素
·当plants[i][j]<target ,则j++,即消去j 元素
·当plants[i][j]==target,则返回true,即找到target
2、若行i下标或者列j下标越界,表明数组中无目标值,返回false
代码:
class Solution {
public boolean findTargetIn2DPlants(int[][] plants, int target) {
int i = plants.length - 1, j = 0;
while(i >= 0 && j < plants[0].length)
{
if(plants[i][j] > target) i--;
else if(plants[i][j] < target) j++;
else return true;
}
return false;
}
}
代码:
class Solution {
public boolean findTargetIn2DPlants(int[][] plants, int target) {
if(plants==null||plants.length==0)
return false;
int row=plants.length; //行数
int col=plants[0].length; //列数
for(int i=row-1,j=0;i>=0&&j<col;){ //从左下角开始查找
if(plants[i][j]==target) //找到
return true;
else if(plants[i][j]>target) //不可能在该行,跳过该行
i--;
else //不可能在该列,跳过该列
j++;
}
return false;
}
}