题目描述:
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解法一:
把每一行都看成有序递增的数组,利用二分查找,通过遍历每一行得到答案,时间复杂度是nlogn。代码如下:
/**
* Created with IntelliJ IDEA.
* Author: 郑文华
* Date: 2017/8/29
* Time: 15:09
* 二分法查找问题
*/
public class Solution {
public boolean Find(int target,int[][] array){
for(int i = 0; i < array.length; i++){
int low = 0;
int high = array[i].length-1;
while(low <= high){
int mid = (low + high) / 2;
if(target>array[i][mid]){
low = mid +1;
}
else if(target<array[i][mid]){
high = mid -1;
}
else
return true;
}
}
return false;
}
}
解法二(最优解):
利用二维数组,由上到下,由左到右递增的规律,那么选区右上角或者左下角的元素array[row][col]与target进行比较,当targrt小于元素array[row][col]时,那么target元素必定在array[row][col]的左边,即col–;当target大于元素array[row][col]时,那么target元素必定在array[row][col]的下边,即row++;
代码如下:
public class Solution {
public boolean Find(int target, int[][] array){
for(int i = 0; i < array.length; i++){
int row = 0;
int col = array[i].length - 1;
while(row < array.length && col >= 0){
if(target == array[row][col]){
return true;
}
else if(target > array[row][col]){
row++;
}
else
col--;
}
}
return false;
}
}