牛客网 二维数组中的查找python实现

来源:互联网 发布:python 函数作为参数 编辑:程序博客网 时间:2024/05/21 09:07

思路:二维数组由上到下,由左到右递增,可以选取右上角或者左下角的元素a[row][col]与target进行比较,

当target小于元素a[row][col]时,target必定在a[row][col]所在行的左边,

当target大于元素a[row][col]时,target必定在a[row][col]所在列的下边

代码实现:

def Find( target, array):    # write code here    n_row = len(array)    n_col = len(array[0])    row = 0    col = n_col - 1    while row < n_row and col >= 0:        if(target == array[row][col]):            return True        elif(target > array[row][col]):            row += 1        else:            col -= 1    return False