57.二叉树的下一个结点(python)

思路

如果该节点有右节点,那么它的下一个结点就是其右子树的最左节点

否则,如果他是父节点的左节点,则返回他的父节点,否则往上找,直到他的某个父节点a是a父节点的左节点,返回a的父节点。

 1 class Solution:
 2     def GetNext(self, pNode):
 3         # write code here
 4         if pNode.right!=None:
 5             tmpNode = pNode.right
 6             while tmpNode.left:
 7                 tmpNode=tmpNode.left
 8             return tmpNode
 9         else:
10             tmpNode = pNode
11             while tmpNode.next:
12                 if tmpNode.next.left == tmpNode:
13                     return tmpNode.next
14                 else:
15                     tmpNode=tmpNode.next
16             return None

2019-12-31 21:53:36

上一篇:60.把二叉搜索树打印成多行(python)


下一篇:24.二叉树中和为某一值的路径(python)