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
}

浙公网安备 33010602011771号