剑指 Offer 04. 二维数组中的查找

和力扣240. 搜索二维矩阵 II 一致
编写一个高效的算法来搜索 _m_ x _n_ 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性:

  • 每行的元素从左到右升序排列。
  • 每列的元素从上到下升序排列。


    示例 1:
    剑指 04. 二维数组中的查找 👌 - 图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

  1. //从右上角遍历,局部剪枝,复杂度O(m+n),空间O1
  2. func searchMatrix(matrix [][]int, target int) bool {
  3. if len(matrix) == 0{
  4. return false
  5. }
  6. row := len(matrix)-1 //横的减少
  7. col:= 0 //纵的增加
  8. for row >= 0 && col < len(matrix[0]) { // 为啥要有 >= 号
  9. if matrix[row][col] == target{
  10. return true
  11. } else if matrix[row][col] > target{
  12. row--
  13. } else if matrix[row][col] < target{ //二分法要注意写else
  14. col++
  15. }
  16. }
  17. return false
  18. }