您的位置:首页 > 其它

二维数组中,每行从左到右递增,每列从上到下递增,给出一个数,判断它是否在数组中

2018-03-10 23:51 369 查看
根据题目我们构造一个二维数组

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)


内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐