根据题目我们构造一个二维数组
matrix = [[1, 2, 3, 4],
[2, 3, 6, 8],
[3, 6, 7, 9]]
由以上构造数组可以看书数据有对角特征,所以我们查找的时候就可以从对角进行,例如从左下角开始遍历数组,我们暂时不考虑相等,如果左下角的元素比给定的数字要大,那么我们就将坐标上移一位,如果上移后又比给定数字小呢,那么我们就将坐标右移一位,不断重复这样的动作。。。
接下来看代码实现
def find_integer(matrix, num):
"""
:param matrix: [[]]
:param num: int
:return: bool
"""
if not matrix:
return False
#获取数组的行列数
rows, cols = len(matrix), len(matrix[0])
row, col = rows - 1, 0#定义初始遍历点
while row >= 0 and col <= cols - 1:
if matrix[row][col] == num:
return True#找到之后直接退出函数,返回True
elif matrix[row][col] > num:#如果遍历点大于给定数值坐标上移
row -= 1
else:#否则坐标右移
col += 1
return False
if __name__ == '__main__':
matrix = [[1, 2, 3, 4],
[2, 3, 6, 8],
[3, 6, 7, 9]]
print 'result of 5 is: ',find_integer(matrix, 5)
print 'result of 6 is: ',find_integer(matrix,6)