二维数组的查找

题目描述

在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

array = [[1,3,5,6],
         [7,8,9,10],
         [11,12,13,14],
         [20,33,44,55]]

def search(target, array):
    row = len(array)-1
    col = len(array[1])-1
    i = row
    j = 0
    while i >= 0 and j <= col:
        if array[i][j] == target:
            return True
        elif array[i][j] < target:
            j += 1
        else:
            i -= 1
    return False
print(search(18,array))