给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:根据树及其中序遍历的特性,来具体考虑各个节点后面的节点值。
public TreeLinkNode GetNext(TreeLinkNode pNode) {
if(pNode == null)return null;
//根据中序遍历的特性,如果一个节点有右子树,那么该节点的下一个节点就为该右子树的最左子节点。
if(pNode.right != null) {
TreeLinkNode cur = pNode.right;
while(cur.left != null){
cur = cur.left;
}
return cur;
//如果没有右子树,如果该节点是父节点的左子节点,那么该节点的下一个节点为其父节点;
//如果该节点为父节点的右子节点,则该节点的下一个节点为该节点向上遍历直至一个节点为其父节点的左子节点的,
//该节点的父节点就为目标节点的下一个节点。
}else if(pNode.next != null) {
TreeLinkNode cur = pNode;
TreeLinkNode parent = pNode.next;
while(parent != null && cur == parent.right) {
cur = parent;
parent = parent.next;
}
return parent;
}
return null;
}