剑指offer—每天一个算法题（二维数组的查找）

 

pyhton代码实现：

"""
题目描述：
    在一个二维数组中（每一个一维数组的长度相同），
    每一行都按照从左到右的顺序递增，
    每一列都按照从上到下的顺序递增，
    请完成一个函数，输入下面的二维数组和一个整数，
    判断数组中是否存在该整数。
            [[ 1  4  7 11 15]
         [ 2  5  8 12 19]
         [ 3  6  9 16 22]
         [10 13 14 17 24]
         [18 21 23 26 30]]
"""
import numpy as np
a = np.array([
    [1,4,7,11,15],
    [2,5,8,12,19],
    [3,6,9,16,22],
    [10,13,14,17,24],
    [18,21,23,26,30]
])
print(a)

def find(target,array):
    """在数组中查找指定元素"""
    if array == []:
        return False
    num_row = len(array)
    num_col =len(array[0])
    
    i,j = 0,num_col - 1
    
    while i >= 0 and j < num_row:
        if array[i][j] > target:
            j -= 1
        elif array[i][j] < target:
            i += 1
        else :
            print(i,j)
            return True

find(5,a)
find(20,a)