第五十六题:二叉树的下一个结点
题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路
① 将给出的二叉树中序遍历的结果保存,在找到目标结点Target的下一个结点
② 结点中包含指向父节点的指针,利用该父指针分析到Target结点的位置上情况(不使用额外空间)
Target结点:
存在右子树: Target结点的右子树最左节点就是Next结点(图中 D B E A C G)
不存在右子树:
Target结点为父节点的左孩子:父节点就是Next结点(图中 N L I)
Target结点为父节点的右孩子:父节点的父节点的…直到当前结点时其父节点的左孩子(图中HJKM)
具体图解如下(注:这里引用牛客网***小河沟大河沟***的图解)
具体实现代码如下
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode){
// 代码的鲁棒性
if(pNode == null){
return pNode;
}
// 右子树不为空
if(pNode.right != null){
return getLeftMost(pNode.right);
}else{
TreeLinkNode parent = pNode.next;
while(parent != null && parent.left != pNode){
pNode = parent;
parent = pNode.next;
}
return parent;
}
}
// 获取最左节点
public TreeLinkNode getLeftMost(TreeLinkNode pNode){
if(pNode == null){
return pNode;
}
while(pNode.left != null){
pNode = pNode.left;
}
return pNode;
}
}
NowCoder(Online Coding, Please Click)