题目:
在一个二维数组中,每一行都按照从左往右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组与一个整数,判断数组中是否含有该整数。
思路:
从二维数组的右上角元素开始进行查找。如果这个元素大于待查找的整数,则从查找范围中去除该元素所在列;若此元素小于待查找整数,则从查找范围中去除该元素所在行。取二维数组剩余部分的右上角元素,再次进行同样的查找过程,直到找到待查找整数或发现二维数组中无此整数。
代码(已在牛客网AC):
public class Solution {
public boolean Find(int target, int [][] array) {
// robust
if(array == null || array.length == 0 || array[0].length == 0 )
return false;
int rowNum = array.length;
int colNum = array[0].length;
int i = 0;
int j = colNum - 1;
while(i <= rowNum - 1 && j >= 0)
{
if(target == array[i][j])
return true;
if(target < array[i][j])
j--;
else
i++;
}
return false;
}
}
性能:
由while循环次数可知,时间复杂度为O(n),空间复杂度为O(1)。