编写一个高效的算法来判断 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
题目分析
根据此矩阵的特性,每一行的第一个元素比前一行的最后一个元素大,并且每一行中元素从小到大排序,所以相当于一个从小到大排列的数字序列存在于二维数组中,要查找目标值的话,首先可以用目标值和第一行最后一个元素比较,如果相等,返回true;如果目标值小于此数,那么就在这一行寻找;如果目标值大于此数,则令行数增加,继续寻找。
代码实现
public boolean searchMatrix(int[][] matrix, int target) { if (matrix.length == 0) return false; int rowIndex = 0; int colIndex = matrix[0].length - 1; while (rowIndex < matrix.length && colIndex >= 0) { if (target == matrix[rowIndex][colIndex]) return true; else if (target > matrix[rowIndex][colIndex]) rowIndex++; else colIndex--; } return false; }
主函数
public static void main(String[] args) { S7 s = new S7(); boolean res1 = s.searchMatrix(new int[][]{{1,3,5,7},{10,11,16,20},{23,30,34,50}}, 3); boolean res2 = s.searchMatrix(new int[][]{{1,3,5,7},{10,11,16,20},{23,30,34,50}}, 13); System.out.println(res1); System.out.println(res2); }
运行结果
true
false