题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
分析
根据二维数组的大小特性来寻找数字 7
:
二维数组的行数为n
,列数为m
,那么:
维护两索引变量,时间复杂度O(n+m)
, 空间复杂度为O(1)
。
代码
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
if not array:
return False
numRow = len(array) - 1
numCol = len(array[0]) - 1
row = 0
col = numCol
while col >= 0 and row<= numRow:
if array[row][col] < target:
row += 1
elif array[row][col] > target:
col -= 1
else:
return True
return False