Data Structures and Algorithm Analysis

杨氏矩阵搜索算法

2018-11-19  本文已影响13人  盗梦者_56f2

介绍

杨氏矩阵中,每行元素是递增的,每列元素也是递增的。即a[i][j]<a[i+1][j]且a[i][j]<a[i][j+1]。要在这样的矩阵中查找某个数值元素的位置,复杂度可以达到O(M+N),其中M为矩阵行长度,N为矩阵列长度。

步骤

  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))
上一篇 下一篇

猜你喜欢

热点阅读