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

题目链接:LeetCode 剑指 Offer 04. 二维数组中的查找

题意:

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

解题思路

可以选择从右上角出发,这样如果比 target 大,则可以剔除当前列,如果比 target 小,可以剔除当前行,直到找到target,如果已经越界了还没有找到,则返回false(也可以从左下角开始)

代码如下:

func findNumberIn2DArray(matrix [][]int, target int) bool {
    if len(matrix) == 0 || len(matrix[0]) == 0 {
        return false
    }
    // 从矩阵的右上角出发
    // 当值比 target 大,剔除当前列,往左走
    // 当值比 target 小,剔除当前行,往下走
    m, n := 0, len(matrix[0])-1
    for m < len(matrix) && n >= 0 {
        if matrix[m][n] > target {
            n--
        } else if matrix[m][n] < target {
            m++
        } else {
            return true
        }
    }
    return false
}
posted @ 2023-07-11 16:41  小星code  阅读(13)  评论(0)    收藏  举报