杨氏矩阵搜索算法

介绍

杨氏矩阵中,每行元素是递增的,每列元素也是递增的。即a[i][j]

步骤

  1. 从矩阵的左下角或者矩阵的右上角处开始递归运行,以左下角为例,value为要查找的值,(i,j)为当前矩阵中的位置,初始为(M-1, 0)。
  2. 如果超过了矩阵范围则说明不存在这样的元素,返回(-1,-1)。
  3. 否则的话,如果当前位置的值大于value,说明要移动位置(向上移一行),使得数值减小,即递归使得i=i-1;如果当前位置的值小于value,说明要移动位置(向右移一列),使得数值增大,即递归使得j=j+1;如果刚好等于value,返回当前的位置(i,j)即可。

python

def young(lst, value):
    raw = len(lst) #行数
    col = len(lst[0]) #列数
    i = raw - 1
    j = 0
    while i >= 0 and j <= col - 1:
        if lst[i][j] == value:
            return (i, j) #找到返回元素位置
        elif lst[i][j] < value:
            j += 1
        else:
            i -= 1
    else:
        return (-1, -1) #没找到

lst = [[1, 3, 5], [2, 4, 6]]
print(yang(lst, 3))

你可能感兴趣的:(杨氏矩阵搜索算法)