*[剑指offer] 二叉树的下一个结点

题目内容

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

https://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e?tpId=13&tqId=11210&tPage=3&rp=3&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

---

题目思路

中序遍历，遍历顺序是左子树+根节点+右子树。现在需要求下一个节点。那么有这些情况：如果包含了右子树，那么就去找右子树中最左侧的节点。如果没有右子树：如果他是某个点的左子树，那么它一个位置一定是它的根节点，如果以上的情况都不满足，那说明它没有右子树，同时是根节点的右节点。（说明它是以某结点R为根节点的树的最后一个节点。）所以它的下一个节点要不然是没有（R是整棵树的根节点），要不然就是一直向上追溯直到碰到这个节点。这个节点的根节点就是所求。

---

程序代码

# -*- coding:utf-8 -*-
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None
class Solution:
    def GetNext(self, pNode):
        # write code here
        if not pNode:
            return None
        if pNode.right:
            pNode=pNode.right
            while pNode.left:
                pNode=pNode.left
            return pNode
        else:
            while pNode.next:
                if pNode=pNode.next.left:#如果该节点是父节点的左侧节点
                    return pNode.next
                pNode=pNode.next#那说明是右节点而且没有右子树，就一直往上跟踪，直到根部或者是某一个节点的左节点