描述
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true
示例 2:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
输出: false
解析
每行有序,且前后行也有序。在最后一列中,找到第一个 >= target的元素,确定target可能在的行。再用简单二分,在这一行查找target即可。
代码
public static boolean searchMatrix(int[][] matrix, int target) { if (null == matrix || matrix.length <= 0 || matrix[0].length <= 0) { return false; } int length = matrix.length; int[] lastCol = new int[length];//最后一列 for (int i = 0; i < length; i++) { lastCol[i] = matrix[i][matrix[0].length - 1]; } int lastColIndex = firstLargeIndex(lastCol, target); if (lastColIndex == -1) { return false; } //简单二分 int[] ponitRow = matrix[lastColIndex];//确定了仅可能在这一行 int start = 0; int end = ponitRow.length - 1; while (start <= end) { int mid = (start + end) >>> 1; if (ponitRow[mid] == target) { return true; } else if (ponitRow[mid] < target) { start = mid + 1; } else { end = mid - 1; } } return false; } /** * 找第一个 >=target的元素位置 * @param nums * @param target * @return */ public static int firstLargeIndex(int[] nums, int target) { int start = 0; int end = nums.length - 1; while (start <= end) { int mid = (start + end) >>> 1; if (nums[mid] < target) { start = mid + 1; } else { end = mid - 1; } } if (start >= 0 && start <= nums.length - 1) { return start; } return -1; }
原文地址:https://www.cnblogs.com/fanguangdexiaoyuer/p/12271648.html
时间: 2024-11-01 10:49:49