LeetCode 二维数组中的查找
题目描述
在一个 n*m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增。请完成一个搞笑的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有改整数。
示例:
现有矩阵 matrix 如下:
1 | [ |
一得之见(Java/Python)
双循环求解。
时间复杂度:O(nm),空间复杂度:O(1)
1 | /** |
1 | def find_number_in_two_dimen_array( |
他山之石(Java/Python)
由于给定的二维数组具备每行从左到右递增以及每列从上到下递增的特点,当访问到一个元素时,可以排除数组中的部分元素。
从二维数组的右上角开始查找。如果当前元素等于目标值,则返回 true。如果当前元素大于目标值,则移到左边一列。如果当前元素小于目标值,则移到下边一行。
可以证明这种方法不会错过目标值。如果当前元素大于目标值,说明当前元素的下边的所有元素都一定大于目标值,因此往下查找不可能找到目标值,往左查找可能找到目标值。如果当前元素小于目标值,说明当前元素的左边的所有元素都一定小于目标值,因为往左查找不可能找到目标值,往下查找可能找到目标值。
- 若数组为空,返回 false
- 初始化行下标为 0,列下标为二维数组的列数减 1
- 重复下列步骤,知道行下标或者列下标超出边界
- 获得当前下标位置的元素 num
- 如果 num 和 target 相等,则返回 true
- 如果 num 大于 target,列下标减 1
- 如果 num 小于 target,行下标加 1
时间复杂度:O(n+m),空间复杂度:O(1)
1 | /** |
效率对比(Java)
1 | 输入: matrix = |
效率对比(Python)
1 | 输入: matrix = |