在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
首先这个题目可以用暴力方法加一定条件的剪枝,但是这样的时间复杂度是 O(mn) ,因为最坏的情况可能会遍历整个二维数组。
还有一种方法是从左下角或者是右上角开始判断,这样每次比较都可以划分出一行或者一列,这时的时间复杂度为 O(m+n) 。
主要说一下第二种算法的思路:
1 首先选择左下角或者右上角是因为每次比较 Matrix[n][m] 后会有三种情况。
假设选择左下角的情况:
等于 target ,这时直接返回 true ;
大于 target ,此时为列的最大值,直接右移比较下一个数;
小于 target ,此时为行的最小值,直接上移比较下一个数。
如果选择其他地方,比如右下角 Matrix[n][m] ,当前值小于 target 时,如果上移会丢失第 n 行的数字,如果左移会丢失第 m 列的数字;其他地方亦会丢失部分数字,所以选择左下角或者右上角。
2 此例搜索 5 ,Matrix[5][5] ,从左下角开始搜索。
1.n = 4 ; m= 0 ; Matrix[4][0] > 5 ; //上移一行
2.n = 3 ; m = 0 ; Matrix[3][0] > 5 ; //上移一行
3.n = 2 ; m = 0 ; Matrix[2][0] < 5 ; //右移一列
4.n = 2 ; m = 1 ; Matrix[2][1] > 5 ; //上移一列
5.n = 1 ; m = 1 ; Matrix[1][1] = 5 ; //返回 true
整理后代码如下:
class Solution {
public boolean findNumberIn2DArray(int[][] matrix, int target) {
if(matrix.length==0||matrix[0].length==0)return false;
int cIndex = 0,rIndex = matrix.length-1;//cindex纵坐标;rindex横坐标;左下角开始遍历
int clength = matrix[0].length;
while(rIndex>=0&&cIndex<clength){
if(matrix[rIndex][cIndex]==target)return true;
else if(matrix[rIndex][cIndex]>target)rIndex--;
else cIndex++;
}
return false;
}
}