golang 算法题 : 二维数组搜索值

Posted 2021-11-29 w3liu

package main

import "fmt"

func main() {
   matrix := [][]int{
      {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},
 }
   exist := searchMatrix(matrix, 5)
   fmt.Println("exit", exist)
}

func searchMatrix(matrix [][]int, target int) bool {
   if matrix == nil || len(matrix) == 0 || len(matrix[0]) == 0 {
      return false
 }
   row := 0
 col := len(matrix[0]) - 1
 for col >= 0 && row <= len(matrix)-1 {
      if matrix[row][col] == target {
         return true
 } else if matrix[row][col] > target {
         col--
      } else if matrix[row][col] < target {
         row++
      }
   }
   return false
}