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

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
}

猜你喜欢

转载自www.cnblogs.com/w3liu/p/10850374.html