杨氏矩阵搜索算法

作者: 盗梦者_56f2 | 来源:发表于2018-11-19 15:07 被阅读13次

    介绍

    杨氏矩阵中,每行元素是递增的,每列元素也是递增的。即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))
    

    相关文章

      网友评论

        本文标题:杨氏矩阵搜索算法

        本文链接:https://www.haomeiwen.com/subject/xuxafqtx.html