编写一个高效的算法来判断 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
思路:
从右上到左下,依次检查当前元素cur,如果target>cur,那么cur所在的列都不用检查了,列向左推进一步。如果target<cur,那么cur所在的行都不用检查了,行向下推荐一步。如果target==cur,返回cur。这个过程是个从右上走到左下的过程。最多走N+M步(N,M是边长),所以,时间复杂度为O(N+M)。
public boolean searchMatrix(int[][] matrix, int target) {
if(matrix.length == 0 || matrix == null) return false;
int row = 0;//行
int column = matrix[0].length; //列
while(row < matrix.length && row >= 0){
if(matrix[row][column] == target){
return false;
}else if(matrix[row][column] > target){
column--;
}else{
row++;
}
}
return false;
}