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