题目描述
描述转载自力扣《74. 搜索二维矩阵》
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
- 每行中的整数从左到右按升序排列。
- 每行的第一个整数大于前一行的最后一个整数。
示例1
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
输出:true
示例2
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
输出:false
解题思路
- 比较 target 和每一行最后一个元素,并截断为一维数组
- 使用二分查找
Java 代码实现
class Solution { public boolean searchMatrix(int[][] matrix, int target) { int rowLen = matrix.length; int colLen = matrix[0].length; int targetRow = 0; for (int i = 0; i < rowLen; ++i) { if (matrix[i][colLen - 1] > target) { targetRow = i; break; } else if(matrix[i][colLen - 1] == target) return true; } int[] arr = matrix[targetRow]; int left = 0, right = colLen - 1, mid = 0; while (left <= right) { mid = (left + right) >> 1; if (arr[mid] == target) return true; else if (arr[mid] > target) right = mid - 1; else left = mid + 1; } return false; } }