二叉树的下一个结点

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

python solution:

# -*- 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):
        root = pNode
        while root.next:
            root = root.next
        self.result = []
        self.midtravel(root)
        return self.result[self.result.index(pNode)+1] if self.result.index(pNode)!=len(self.result)-1 else None
        
        
    def midtravel(self,root):
        if not root:
            return 
        self.midtravel(root.left)
        self.result.append(root)
        self.midtravel(root.right)