LeetCode(python):剑指 Offer 04. 二维数组中的查找
1. 题目:剑指 Offer 04. 二维数组中的查找
2. 题目链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/
3. 题目内容
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
4. 输入输出示例
输入:
[ [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
5. 题目分析
从[0, col]开始,由于这个值是这一行最大的一个值同时向右边和下面进行比较,记录right和down都为0如果target比左边的大,right则减1如果target比左边的小,down则加1如果遇到target相等的值,则return true如果左边和下面都比target大了还没有找到相等的,就结束循环注意临界值
6. python代码
class Solution(object):
def findNumberIn2DArray(self
, matrix
, target
):
"""
解题方案:
1. 从[0, col]开始,由于这个值是这一行最大的一个值
2. 同时向右边和下面进行比较,记录right和down都为0
3. 如果target比左边的大,right则减1
4. 如果target比左边的小,down则加1
5. 如果遇到target相等的值,则return true
6. 如果左边和下面都比target大了还没有找到相等的,就结束循环
7. 注意临界值
:type matrix: List[List[int]]
:type target: int
:rtype: bool
"""
if not matrix
or not matrix
[0]:
return False
row
= len(matrix
)
col
= len(matrix
[0])
if matrix
[0][0] > target
:
return False
elif matrix
[0][0] == target
:
return True
if row
< 4 and col
< 4:
for i
in range(row
):
for j
in range(col
):
if matrix
[i
][j
] == target
:
return True
return False
if row
== 1:
return bool(target
in matrix
[0])
if col
== 1:
for i
in range(row
):
if matrix
[i
][0] == target
:
return True
return False
down
, right
= 0, col
-1
while down
< row
and right
>= 0:
num
= matrix
[down
][right
]
if num
== target
:
return True
elif num
> target
:
right
-= 1
else:
down
+= 1
return False
7. 其他解法
暴力法,时间复杂度O(n*m)由于每一行都是顺序排列的,很容易想到二分查找法,遍历每一行,对美一行进行二分查找,时间复杂读O(n*logn)