二叉树的下一个节点

题目描述

给定一个二叉树其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示，从子节点指向父节点的用虚线表示

示例:
输入:{8,6,10,5,7,9,11},8
返回:9
解析:这个组装传入的子树根节点，其实就是整颗树，中序遍历{5,6,7,8,9,10,11}，根节点8的下一个节点就是9，应该返回{9,10,11}，后台只打印子树的下一个节点，所以只会打印9，如下图，其实都有指向左右孩子的指针，还有指向父节点的指针，下图没有画出来

输入描述：
输入分为2段，第一段是整体的二叉树，第二段是给定二叉树节点的值，后台会将这2个参数组装为一个二叉树局部的子树传入到函数GetNext里面，用户得到的输入只有一个子树根节点
返回值描述：
返回传入的子树根节点的下一个节点，后台会打印输出这个节点

输入输出

示例1
输入：{8,6,10,5,7,9,11},8
返回值：9

示例2
输入：{8,6,10,5,7,9,11},6
返回值：7

示例3
输入：{1,2,#,#,3,#,4},4
返回值：1

示例4
输入：{5},5
返回值：“null”

说明：
不存在，后台打印"null"

解题思路

1. 找当前节点的右子树，如果存在，就一直找右子树最左边就是下一节点
2. 如果不存在右子树，找当前节点的父节点，看当前节点是不是父节点的左子树，一直找到它是父节点的左子树，大于父节点

代码示例

# -*- 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 pNode.right:
            tempNode = pNode.right
            # 寻找右子树的左节点,直到找到最后的左节点
            while tempNode.left:
                tempNode = tempNode.left
            return tempNode
        # 如果不存在右子树，找其父节点
        else:
            tempNode = pNode
            # 父节点存在 直到找到一个其属于左子树的父节点，
            while tempNode.next:
                # 找到父节点的左子树也是当前节点，则返回父节点
                if tempNode.next.left == tempNode:
                    return tempNode.next
                # 如果当前节点父节点的左子树不是该节点，
                # 则调整节点，找父节点的父节点
                tempNode = tempNode.next
            return None