面试题3. 二维数组中的查找
原创
©著作权归作者所有:来自51CTO博客作者mb6304a73bed12a的原创作品,请联系作者获取转载授权,否则将追究法律责任
题目描述
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路:
首先选取数组右上角的元素:
- 如果这个数 = 要查找的数,直接返回true,程序完毕
- 如果这个数 < 要查找的数,说明这个数左边的数都小于要查找的数,则这一行都不要了
- 如果这个数 > 要查找的数,说明这个数下边的数都大于要查找的数,则这一列都不要了
以此类推,最后找到要查找的数。
对于有n个元素的矩阵,时间复杂度要小于O(n)
Java代码如下:
public class Solution {
public boolean Find(int target, int [][] array) {
int i = 0;
int j = array[0].length - 1;
while (i < array.length && j >= 0) {
if (target < array[i][j]) {
j--;
} else if (target > array[i][j]) {
i++;
} else {
return true;
}
}
return false;
}
}