剑指 Offer 04. 二维数组中的查找
和力扣240. 搜索二维矩阵 II 一致
编写一个高效的算法来搜索 _m_ x _n_
矩阵 matrix
中的一个目标值 target
。该矩阵具有以下特性:
- 每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
示例 1:
输入:matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 5
输出:true
//从右上角遍历,局部剪枝,复杂度O(m+n),空间O1
func searchMatrix(matrix [][]int, target int) bool {
if len(matrix) == 0{
return false
}
row := len(matrix)-1 //横的减少
col:= 0 //纵的增加
for row >= 0 && col < len(matrix[0]) { // 为啥要有 >= 号
if matrix[row][col] == target{
return true
} else if matrix[row][col] > target{
row--
} else if matrix[row][col] < target{ //二分法要注意写else
col++
}
}
return false
}