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

题目

在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
示例：

现有矩阵 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。
给定 target = 20，返回 false。

---

解决方案：二叉树

利用二叉树的思想，从左下角向右上角遍历，或者反过来，从右上角向左下角遍历

//从左下角向右上角遍历
func findNumberIn2DArray(matrix [][]int, target int) bool {
    //以左下角为起点
    i := len(matrix)-1//获取左下角y坐标
    j := 0//获取左下角x坐标
    for i>-1 && j<len(matrix[i]){       
        if target<matrix[i][j]{
            i--  //小于target,向上查找
        }else if target>matrix[i][j]{
            j++ //大于targat,向右查找
        }else if target==matrix[i][j]{
            return true
        }
    }
    return false//超出matrix返回false
}

//从右上角向左下角遍历
func findNumberIn2DArray(matrix [][]int, target int) bool {
    if len(matrix) == 0 || len(matrix[0]) == 0 {
		return false
	}
    //以右上角为起点
    i := 0//获取右上角y坐标
    j := len(matrix[0])-1//获取右上角x坐标
    for j>-1 && i<len(matrix){       
        if target<matrix[i][j]{
            j--  //大于target,向左查找
        }else if target>matrix[i][j]{
            i++ //大于targat,向下查找
        }else if target==matrix[i][j]{
            return true
        }
    }
    return false//超出matrix返回false
}

---

解决方案：二重遍历

// for-range 遍历
func findNumberIn2DArray(matrix [][]int, target int) bool {
    for _, value1 := range matrix{
        for _,value2 := range value1{
            if value2 == target{
                return true
            }
        }
    }
    return false
}

//普通遍历
func findNumberIn2DArray(matrix [][]int, target int) bool {
    for i:= 0 ; i < len(matrix); i++ {
        for j :=0; j < len(matrix[i]); j++ {
            if target == matrix[i][j] {
                return true
            }
        }
    }
    return false
}