剑指offer----二维数组中的查找

有序二维数组,数值从左到右递增,从上到下递增,查找某一个数是否存在
思路:首先选取数组中右上角的数字,如果该数字等于要查找的数字,则查找过程结束;如果该数字大于要查找的数字,则剔除该数字所在的列;如果该数字小于要查找的数字,则剔除这个数字所在的行。

def findNum(array,target):
    rows=len(array)
    cols=len(array[0])
    row=0
    col=cols-1
    while row<rows and col>=0:
        if array[row][col]==target:
            return True
        elif array[row][col]>target:
            col=col-1
        else:
            row=row+1
    return False

if __name__=="__main__":
    target=10
    array=[[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]]
    print(findNum(array,target))

python中数组定义方式
方式一:直接定义

array=[[0,0,0],[0,0,0],[0,0,0]]

方式二:间接定义

array=[[0 for i in range(3)] for i in range(3)]

猜你喜欢

转载自blog.csdn.net/haoshan4783/article/details/88682359