剑指offer：二叉树的下一个节点（Python）

题目描述

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

解题思路

首先要清楚的是，现在这个二叉树的节点样式不仅包含左右子节点，还包括它的父节点，样式如下。其中 self.next 指向该节点的父节点。

class TreeLinkNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
        self.next = None

这道题意即：给定一个节点，按照中序遍历（左根右）的方式求该节点的下一个节点。有三种情况：
1. 给定的节点为空——返回空；
2. 给定的节点有右子树——沿着该右子树，返回右子树的第一个左叶子节点；
3. 给定的节点没有右子树——如果位于某个节点的左子树中，则上溯直至找到该节点；否则就返回空。【因为按照中序遍历“左中右”的遍历方式，当该节点没有右子树时，要么遍历完毕，下一个节点为空；要么某个子树的左子树遍历完毕，下一个节点是子树的根节点】

class Solution:
    def GetNext(self, pNode):
        if not pNode:
            return
        if not pNode.right:
            return self.getHead(pNode)
        return self.getRightNode(pNode.right)

    def getHead(self, pNode):
        while pNode.next:
            if pNode.next.right == pNode:
                pNode = pNode.next
            elif pNode.next.left == pNode:
                return pNode.next
        return None

    def getRightNode(self, pNode):
        while pNode.left:
            pNode = pNode.left
        return pNode

以上是实现的代码。本来不想实现一个题目用三个函数的，但是解完题后失去了合并成一个函数的动力，因为个人认为这种方式更直观，更便于理解。写完之后也没有找到更简短精妙的实现方法，就这样啦~~