题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
解题思路
leetcode上好像没有相关的题目,是在牛客上做的。首先要看一下树结点的结构,是具备左右结点的指针以及指向父节点的指针(next结点)的。这就使得我们在中序遍历时可以追溯二叉树要遍历的下一个结点。
那我们中序遍历的时候是怎么做的?左-中-右。现在给到一个结点pNode,它可以是在左中右这三个位置上,所以需要具体讨论。
首先第一个情况:结点作为根节点,此时它有右子树的话,那下一个结点就是右子树的最左叶结点了。
第二个情况:假如结点没有右子树,那它就得往父结点上面找下一个结点了。
有两种情况可以具体分析一下(可以画画图),首先它可能是作为某个结点的一个右结点,并拥有一颗左子树,这时它的下一个结点是哪里呢?它的父结点显然已经被遍历过了,所以下一个结点应该是它的父节点的某个祖宗结点(也就是向上追溯,直到有一个父结点是作为树的左节点)。
另一种情况,它可能恰好就是某颗树的左节点,那它的下一个结点就是它的父亲,这时候就满足pNode===pNode.next.left,可以直接返回父节点了。
Code
/*function TreeLinkNode(x){
this.val = x;
this.left = null;
this.right = null;
this.next = null;
}*/
function GetNext(pNode)
{
// write code here
// 中序遍历,首先获取到根节点,接下来就要去判断右子树
if(!pNode) return null;
// 情况一:右子树非空,取右子树的最左子节点
if(pNode.right){
pNode=pNode.right;
while(pNode.left){
pNode=pNode.left;
}
return pNode;
}
// 情况二:右子树空,判断当前节点是否是父节点的左节点
// 是的话就返回它的父节点作为下一个结点
while(pNode.next){
if(pNode===pNode.next.left){
return pNode.next;
}
// 不满足,则通过next回溯父节点,直到有一个结点等于其父节点的左节点
pNode=pNode.next;
}
return null;
}
运行环境:JavaScript (V8 6.0.0)
运行时间:14ms
占用内存:11692k