题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
public class Solution {
/*
* 我的答案
*/
public static boolean Find(int target, int[][] arr) {
// 二分查找
int arrLen = arr.length;
int colLen = arr[0].length;
if (arrLen == 0 || colLen == 0) {
return false;
}
if (target < arr[0][0] && target > arr[arrLen - 1][colLen - 1]) {
return false;
}
boolean res = false;
for (int i = 0; i < arrLen; i++) {
// 符合在区间内部才进行二分查找
if (target >= arr[i][0] && target <= arr[i][colLen - 1]) {
// 2.在一维数组内部进行二分
res = binarySearch(0, colLen - 1, arr[i], target);
if (res) {
return true;
}
}
}
return res;
}
/*
* 这个更好,更好的答案,非递归方式更简洁
*
* 把每一行看成有序递增的数组, 利用二分查找, 通过遍历每一行得到答案, 时间复杂度是nlogn
*/
public static boolean Find2(int target, int[][] arr) {
int left = -1;
int right = -1;
int mid = -1;
for (int i = 0; i < arr.length; i++) {
left = 0;
right = arr[i].length - 1;
while (left <= right) {
mid = (left + right) / 2;
if (target < arr[i][mid]) {
right = mid - 1;
} else if (target > arr[i][mid]) {
left = mid + 1;
} else {
return true;
}
}
}
return false;
}
public static boolean binarySearch(int start, int end, int[] arr, int target) {
if (start > end) {
return false;
}
int mid = (start + end) / 2;
if (arr[mid] < target) {
start = mid + 1;
return binarySearch(start, end, arr, target);
} else if (arr[mid] > target) {
end = mid - 1;
return binarySearch(start, end, arr, target);
} else {
return true;
}
}
}