二叉树下一个节点_二叉树一个节点-CSDN博客

本文链接：https://blog.csdn.net/qq_40146935/article/details/109067925

分4种情况：
当该节点存在右子树时，那么下一个节点一定在右子树中产生
这种情况下又分成两种情况：
（1）右子树不存在左分支，那么要找的下一个节点就是右子树的根节点。
（2）右子树中存在左分支，那么不停的寻找左分支的左分支就可以了。
2.当该节点不存在右子树时，那么下一个节点一定在该节点的父节点中产生，这种情况下又分两种情况：
（1）该节点是他的父节点的左孩子，那么下一个节点就是该节点的父节点
（2）该节点是他父节点的右孩子，那么就需要继续往该节点的父节点的父节点去寻找了。然后重复前面的分析。

class TreeNode():
    def __init__(self,data):
        self.data = data
        self.left = None
        self.right = None
        
class Solution():
    def get_next_node(self,pNode):
        if not pNode:
                return None
        if pNode.right:
            pNode = pNode.right
            while pNode.left:
                pNode = pNode.left
            return pNode
        else:
            while pNode.parent:
                if pNode == pNode.parent.left:
                    return pNode.parent
                pNode = pNode.next
        return None