剑指offer之二叉树的下一个节点

题目描述

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

 tips：给定当前节点，求其在中序遍历的下一个节点，并且已知每个节点的父节点。

方法一，可以通过给定节点的父节点循环追溯至根节点，然后通过根节点进行中序遍历，找到给定节点的下一个节点。该方法需要遍历整棵树，复杂度较高。

方法二，我们只需遍历有限的几个分支即可找到下一个节点。中序遍历即按照左子树、根节点、右子树的顺序遍历。那么

①如果当前节点有右子树，下一个节点为右子树的最左边的叶子节点

②如果当前节点无右子树，向父节点追溯，若当前节点是父节点的左孩子，输出父节点。否则继续向上追溯，更新父节点为当前节点，继续判断当前节点与父节点的关系。边界判断，若追溯到根节点还未输出，继续循环，一般根节点的父节点为空，故无需单独判定边界，直接输出父节点即可。

代码如下：

# -*- 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
        if pNode.right:
            pNode=pNode.right
            while pNode.left:
                pNode=pNode.left
            return pNode
        else:
            parent=pNode.next
            while parent and parent.left!=pNode:
                parent,pNode=parent.next,parent
            return parent