二维数组中的查找
题目描述:
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路:
由于原题说到,二维数组按行有序且按列有序,如果从矩阵的四个角出发,可以通过按行或者按列的方法找到目标值。本题中,我们以右上角为例,通过与目标值的比较,确定是向左移动还是向下移动。
解题代码:
class Solution:
# array 二维列表
def Find(self, target, array):
#判断输入是否合法
if not array:
return None
row_length = len(array)
col_length = len(array[0])
#将当前坐标位置设为右上角
cur_row = 0
cur_col = col_length - 1
#在合法边界内进行比较
while (cur_row <= row_length-1 and cur_col >= 0):
#如果与目标相等 返回
if array[cur_row][cur_col] == target:
return True
#如果小于目标值,向下一行
elif array[cur_row][cur_col] < target:
cur_row += 1
#如果大于目标值,向左一列
else:
cur_col -= 1
return False