题目描述
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例:
输入: matrix = [ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ] target = 3 输出: true
思路
1.我们可以从二维数组的右上角开始搜索,若target大于当前元素,我们的检索方向就向下,若小于,我们的检索方向就向左,若检索出界,还没找到结果,直接返回false即可。
Java代码实现
public boolean searchMatrix(int[][] matrix, int target) { if(matrix.length == 0) return false; int startRow = 0; int startCol = matrix[0].length-1; while(startRow<matrix.length && startCol>=0){ if(target > matrix[startRow][startCol]){ //向下走 startRow++; }else if(target<matrix[startRow][startCol]){ //向左走 startCol--; }else{ return true; } } return false; }
Golang代码实现
func searchMatrix(matrix [][]int, target int) bool { if len(matrix) == 0{ return false } row := 0 col := len(matrix[0])-1 for row< len(matrix) && col >= 0{ cur := matrix[row][col] if cur == target{ return true }else if cur > target{ col-- }else{ row++ } } return false }