二维数组中的查找
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路
因为二维数组递增的关系,数组中的数字左边和上边的数字都比它小,右边和下边的数字都比它大。可以从左下角或右上角开始查找,以右上角开始为例,当前数字若比要查找的整数n小,则行+1;若比n大,则列-1;若相等,返回true。
python代码
class Solution:
# array 二维列表
def Find(self, target, array):
if not len(array) or not len(array[0]): #判断列表是否为空
return False
#定义开始位置为列表左下角
row = len(array) - 1
col = 0
#设置边界条件,开始查找target
while row >= 0 and col < len(array[0]):
if target == array[row][col]:
return True
#当前数字小于target,则列+1
elif target > array[row][col]:
col += 1
#当前数字大于target,则行-1
else:
row -= 1
return False