#建立一个二叉树
#声明一个二叉树节点类
#节点中包括:节点值,节点左子树根节点,节点右子树根节点,节点父节点
class TreeNode():
def __init__(self,value=None,leftNode=None,rightNode=None,parentNode=None):
self.val = value
self.leftnode = leftNode
self.rightnode = rightNode
self.parentnode = parentNode
def SetLeftNode(self,leftNode):
self.leftnode = leftNode
def SetRightNode(self,rightNode):
self.rightnode = rightNode
def SetParentNode(self,parentNode):
self.parentnode = parentNode
def GetLeftNode(self):
return self.leftnode
def GetRightNode(self):
return self.rightnode
def GetParentNode(self):
return self.parentnode
def GetValue(self):
return self.val
#定义根节点以及各个节点
root = TreeNode(1)
n2 = TreeNode(2)
n3 = TreeNode(3)
n4 = TreeNode(4)
n5 = TreeNode(5)
n6 = TreeNode(6)
n7 = TreeNode(7)
#设置根节点的左右节点
root.SetLeftNode(n2)
root.SetRightNode(n3)
#设置n2节点左右子树节点和其父节点
n2.SetLeftNode(n4)
n2.SetRightNode(n5)
n2.SetParentNode(root)
#设置n3节点左右子树节点和其父节点
n3.SetLeftNode(n6)
n3.SetRightNode(n7)
n3.SetParentNode(root)
#设置n4,n5节点的父节点
n4.SetParentNode(n2)
n5.SetParentNode(n2)
#设置n6,n7节点的父节点
n6.SetParentNode(n3)
n7.SetParentNode(n3)
#给了任意一个结点,输出中序遍历时,其下一个节点所存储数值
#节点中序遍历下一个节点有两种情况:
#1、其为它的右子树的左子树的最左子节点
#2、其为它的父节点的左孩子时候的根节点
def GetNextNode(node):
if node.GetRightNode() != None:
node = node.GetRightNode()
while node.GetLeftNode()!=None:
node = node.GetLeftNode()
return node.GetValue()
else:
while node.GetParentNode() != None:
parentNode = node.GetParentNode()
if parentNode.GetLeftNode() == node:
return parentNode.GetValue()
node = parentNode
return None
#这里输出测试n3节点在中序遍历时候的下一个节点->n7(值为7)
print(GetNextNode(n3))
#这里输出测试n5节点在中序遍历时候的下一个节点->n1(值为1)
print(GetNextNode(n5))