描述
给定一个二叉树其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。
注意,树中的结点不仅包含左右子结点,同时包含指向父结点的next指针。
基本思路
分三种情况:
1、有右子树,则它的下一个节点是右子树的最左孩子
2、无右子树,且它是左孩子叶子节点,则返回当前节点的父节点
3、无右子树,且它是右孩子叶子节点,则一直向上找,直到某个上级是上上级的左孩子,则返回这个上上级
/*function TreeLinkNode(x){
this.val = x;
this.left = null;
this.right = null;
this.next = null;
}*/
function GetNext(pNode)
{
// write code here
//在节点有右子树的情况下,就打印右子树的最左节点
if(pNode.right){
let node = pNode.right
while(node.left){
node = node.left
}
return node
}
//节点无右子树
//如果父节点已经是null,则说明此节点为根,则直接在下面返回null
//如果还有父节点
while(pNode.next){
//判断当前节点是否为父节点的左孩子节点且是叶子节点,如果是,则返回当前节点的父节点
if(pNode.next.left === pNode) return pNode.next
//叶子节点且是右孩子,则一直向上寻找,直到它的上级是上上级的左孩子,打印了这个上上级
pNode = pNode.next
}
return null
}
module.exports = {
GetNext : GetNext
};