1. 程式人生 > >劍指offer————在二維數中查詢(python)

劍指offer————在二維數中查詢(python)

題目描述

在一個二維陣列中(每個一維陣列的長度相同),每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函式,輸入這樣的一個二維陣列和一個整數,判斷陣列中是否含有該整數。

def f(a,n):
    c,r = len(a)-1,len(a[0])-1
    if a[0][0] > n or a[c][r] < n:
        return 'False'
    i,j = 0,r
    while i != r+1 and j != -1:
        if a[i][j] == n:
            return i+1,j+1
        elif a[i][j] < n:
            i += 1
        else:
            j -= 1
    return 'False'
a = [[1,2,8,9],
    [2,4,9,12],
    [4,7,10,13],
    [6,8,11,15]]
f(a,4)