问题描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路
分情况讨论:
1.如果当前节点的右子树不为空,则下一个节点为当前节点右子树上的最左节点。
2.如果当前节点的右子树为空,则需要向上寻找:从当前节点开始,寻找一个祖先节点,这个节点是父节点的左子节点,那么答案就是这个父节点。如果一直走到null都没有这样的节点和父节点,那么说明当前节点没有中序遍历的后继节点,返回null。
代码实现
/*
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
if(pNode == null){
return null;
}
//如果当前节点的右子树不为空,下一个节点在则是右子树上的最左节点
//如果当前节点的右子树为空,则从当前节点一直向上遍历,直到找到一个节点是父节点的左子节点,答案就是这个父节点
//如果找不到,则说明当前节点没有中序遍历的后继节点
if(pNode.right != null){
TreeLinkNode node = pNode.right;
while(node.left != null){
node = node.left;
}
return node;
}else{
TreeLinkNode node = pNode;
TreeLinkNode parent = node.next;
while(parent != null){
if(parent.left == node){
return parent;
}
node = parent;
parent = parent.next;
}
if(parent == null){
return null;
}
}
return null;
}
}