04.二维数组中的查找

OF4.二维数组中的查找 #

题目描述 #

题目地址

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

示例 1 #

输入
[
  [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 : 线性查找 #

由于给定的二维数组具备每行从左到右递增以及每列从上到下递增的特点,当访问到一个元素时,可以排除数组中的部分元素。

从二维数组的右上角开始查找。如果当前元素等于目标值,则返回 true。如果当前元素大于目标值,则移到左边一列。如果当前元素小于目标值,则移到下边一行。

可以证明这种方法不会错过目标值。如果当前元素大于目标值,说明当前元素的下边的所有元素都一定大于目标值,因此往下查找不可能找到目标值,往左查找可能找到目标值。如果当前元素小于目标值,说明当前元素的左边的所有元素都一定小于目标值,因此往左查找不可能找到目标值,往下查找可能找到目标值。

算法流程:

  1. 若数组为空,返回 false

  2. 初始化行下标为 0,列下标为二维数组的列数减 1

  3. 重复下列步骤,直到行下标或列下标超出边界

    1. 获得当前下标位置的元素 num
    2. 如果 num 和 target 相等,返回 true
    3. 如果 num 大于 target,列下标减 1
    4. 如果 num 小于 target,行下标加 1
  4. 循环体执行完毕仍未找到元素等于 target ,说明不存在这样的元素,返回 false`

    作者:

复杂度分析:

  • 时间复杂度$$O(M+N)$$**:**访问到的下标的行最多增加 n 次,列最多减少 m 次,因此循环体最多执行 n + m
  • 空间复杂度$$O(1)$$

代码 #

{% tabs %} {% tab title=“Go” %}

func findNumberIn2DArray(matrix [][]int, target int) bool {
	if len(matrix) == 0 {
		return false
	}
	rLen := len(matrix)
	cLen := len(matrix[0])

	//	从右上角的元素找起来
	for r, c := 0, cLen-1; r < rLen && c >= 0; {
		if matrix[r][c] == target {
			return true
		}
		if matrix[r][c] > target {
			c--
		} else {
			r++
		}
	}
	return false
}

{% endtab %} {% endtabs %}

#

总结 #

如果你同我一样热爱数据结构、算法、LeetCode,可以关注我 GitHub 上的 算法 题解: awesome-golang-algorithm