剑指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)