介绍
杨氏矩阵中,每行元素是递增的,每列元素也是递增的。即a[i][j]<a[i+1][j]且a[i][j]<a[i][j+1]。要在这样的矩阵中查找某个数值元素的位置,复杂度可以达到O(M+N),其中M为矩阵行长度,N为矩阵列长度。
步骤
- 从矩阵的左下角或者矩阵的右上角处开始递归运行,以左下角为例,value为要查找的值,(i,j)为当前矩阵中的位置,初始为(M-1, 0)。
- 如果超过了矩阵范围则说明不存在这样的元素,返回(-1,-1)。
- 否则的话,如果当前位置的值大于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))
网友评论