题目logo-dark-cn.c42314a8.svg

在一个 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。


解决方案:二叉树

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

  1. //从左下角向右上角遍历
  2. func findNumberIn2DArray(matrix [][]int, target int) bool {
  3. //以左下角为起点
  4. i := len(matrix)-1//获取左下角y坐标
  5. j := 0//获取左下角x坐标
  6. for i>-1 && j<len(matrix[i]){
  7. if target<matrix[i][j]{
  8. i-- //小于target,向上查找
  9. }else if target>matrix[i][j]{
  10. j++ //大于targat,向右查找
  11. }else if target==matrix[i][j]{
  12. return true
  13. }
  14. }
  15. return false//超出matrix返回false
  16. }
//从右上角向左下角遍历
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
}