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

Posted 2020-11-25 Assange

思路

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

否则，如果他是父节点的左节点，则返回他的父节点，否则往上找，直到他的某个父节点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