题目:给定一颗二叉树和其中的一个节点,如何找到中序遍历序列的下一个节点?树中的节点除了有两个分别指向左、右子节点的指针,还有一个指向父节点的指针。
我们就以上图中的二叉树为例,它的中序遍历是{4, 2, 8, 5, 9, 1, 6, 3, 7}。分析某个节点的下一个节点位置我们可以得到三种情况。
(设:输入节点为node,它的下一节点为nextNode)
情况一:如果node有右子节点,那么nextNode就是node的右子节点的最左子节点。
情况二:如果node没有右子节点,并且node是它父节点的右子节点,这种情况我们就沿着父节点向上遍历,直到遇到一个节点的左子节点是父节点的节点,这个节点就是nextNode。
情况三:如果node没有右子节点,并且node是它父节点的左子节点,这种情况最简单,nextNode就是它的父节点。
三种情况分析完后就很容易写出下面的代码了。
public void findNextNode(BinaryTreeNode node) {
if (node != null) return;
BinaryTreeNode nextNode = null;
if (node.rightNode != null) { // node有右子节点
BinaryTreeNode right = node.rightNode;
// 循环找到右子节点的最左子节点
while (right.leftNode != null) {
right = right.leftNode;
}
nextNode = right;
} else if (node.parentNode != null) { // node没有右子节点但有父节点
BinaryTreeNode current = node;
BinaryTreeNode parent = node.parentNode;
// 沿着父节点向上遍历,直到遇到一个节点的左子节点是父节点的节点
while (parent != null && current.equals(parent.rightNode)) {
current = parent;
parent = parent.parentNode;
}
nextNode = parent;
}
if (nextNode != null) {
System.out.println("nextNode's value is " + nextNode.value);
}
}
画图分析规律和可能性很重要!