二叉树的下一个节点

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

思路:1.二叉树为空,返回空
2.# 节点右孩子存在,则设置一个指针从该节点的右孩子出发,一直沿着指向左孩子的指针找到的没有左孩子的节点(不一定为叶子节点,也可以有右孩子)即为下一个节点
3. # 节点不是根节点,(因为上面讨论了节点有孩子存在,这里节点右孩子都不存在)若该节点为其父节点的左孩子,则返回父节点,否则继续向上遍历,直到找到其最近父节点,如果其最近父节点是其最近父节点的父节点的左孩子,返回最近父节点的父节点。重复之前的判断

# -*- 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):
		if not pNode:
			return None
		if pNode.right:
			pNode = pNode.right
			while pNode.left:
				pNode = pNode.left
			return pNode
		while pNode.next:
			proot = pNode.next
			if proot.left == pNode:
				return proot
			pNode = pNode.next  **# 注意这里不是pNode = pNode.next.next,**例:只有{4,2,5,1,3}在节点为5时,下一个节点本为1,因为pNode = pNode.next.next为1,而proot为None 造成错误
		return None
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值