1 问题
在一个二维数组array中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
[
[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]
]
给定 target = 7,返回 true。
给定 target = 3,返回 false。
数据范围:矩阵的长宽满足 0≤n,m≤500 , 矩阵中的值满足 0≤val≤10^9
进阶:空间复杂度 O(1) ,时间复杂度 O(n+m)
示例1
输入:7,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]
返回值:true
说明:存在7,返回true
示例2
输入:1,[[2]]
返回值:false
示例3
输入:3,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]
返回值:false
说明:不存在3,返回false
2 答案
自己写的,但是好像不符合时间复杂度 O(n+m),没有利用到矩阵内部的行列都是有序这个性质
class Solution:
def Find(self , target: int, array: List[List[int]]) -> bool:
# write code here
for i in range(len(array)):
for j in range(len(array[0])):
if array[i][j] == target:
return True
return False
官方解,二分查找
- step 1:首先获取矩阵的两个边长,判断特殊情况。
- step 2:首先以左下角为起点,若是它小于目标元素,则往右移动去找大的,若是他大于目标元素,则往上移动去找小的。
- step 3:若是移动到了矩阵边界也没找到,说明矩阵中不存在目标值。
class Solution:
def Find(self, target: int, array: List[List[int]]) -> bool:
# 优先判断特殊
if len(array) == 0:
return False
m = len(array)
if len(array[0]) == 0:
return False
n = len(array[0])
i = m - 1
j = 0
# 从最左下角的元素开始往左或往上
while i >= 0 and j < n:
if target > array[i][j]:
j += 1
elif target < array[i][j]:
i -= 1
else:
return True
return False
https://www.nowcoder.com/share/jump/9318638301698255316940