二叉树的下一个结点
题目
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思想
分三种情况:
1.如果给定结点有右孩子,下一个结点是右孩子的最左孩子。
2.如果给定结点无右孩子且为根节点的左孩子,下一个是根节点。
3.如果给定结点无右孩子且为根节点的右孩子,则找到其为左分支的根节点,下一个为根。
代码
(注意边界条件。)
public static TreeLinkNode getNext(TreeLinkNode pNode) {
//如果树为空,返回空
if (pNode == null)
return null;
TreeLinkNode pNext = null;
//如果有右孩子,找到有孩子的最左孩子
if (pNode.right != null) {
TreeLinkNode pRight = pNode.right;
while (pRight.left != null)
pRight = pRight.left;
pNext = pRight;
//如果没有右孩子
} else if (pNode.next != null) {
TreeLinkNode pParent = pNode.next;
TreeLinkNode pCurrent = pNode;
if (pCurrent == pParent.left)
pNext = pParent;
//他是根节点的右孩子,找到他为左分支的根节点
while (pParent != null && pCurrent == pParent.right) {
pCurrent = pParent;
pParent = pParent.next;
}
pNext = pParent;
}
return pNext;
}