剑指 Offer 04. 二维数组中的查找
题目描述:
在一个 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。
解1:最先想到的办法就是遍历二维数组,优点就是简单易实现,但是时间复杂度高
public static boolean findNumberIn2DArray(int[][] m, int target) {//时间复杂度高
for(int i=0;i<m.length;i++) //效率太低
{
for(int j=0;j<m[i].length;j++) {
if(m[i][j]==target)return true;
}
}
return false;
}
解2:因为这个数组很有规律,它是行从左到右递增,列从上到下递增,所以可以 *从二维矩阵的左下角或者是右上角开始遍历, 如果从右上角开始,如果t小于该值,t小于这一列,这一列都可以排除,左移一列,如果t大于该值,t大于这一行,当前这行都可以排除,下移一行。
以 target = 5为例:
从右上角开始
target <15,第四列排除,左移一列,变成
target<11,j继续左移
target<7,左移
target>4,下移
找到target,结束
Java代码如下:
public static boolean findNumberIn2DArray(int[][] m, int t) {
if (m == null || m.length == 0 || m[0].length == 0) {
return false;
}//!!
int row=m.length;
int col=m[0].length;
int i=0,j=col-1;
while(j>=0&&i<row)
{
if(m[i][j]>t) {
j--;
}
else if(m[i][j]<t) {
i++;
}
else {
return true;
}
}
return false;
}