题目:给定一棵二叉树和其中的一个节点,如何找出中序遍历序列的下一个节点?书中的节点除了有两个分别指向左右子节点的指针,还有一个指向父节点的指针
此题在 LeetCode 中没有,我想是因为没法实现指向父节点的指针,但是我们可以写一个函数找到父亲节点,这都不重要,方法是最重要的:
当某个节点存在右孩子的时候,那该节点的下一个节点就是右孩子的最左边的孩子,类似于寻找二叉搜索树右子树的最小值,如图节点 2,该节点存在右子树,那么中序遍历下一个节点就是 4 的最左子树,也就是 8,是 2 的下一个节点
如果该节点不存在右孩子,那就看他是父亲节点的左孩子还是右孩子,如果是左孩子,那么下一个节点就是父亲节点,比如节点 1, 0 ,下一个节点都是他们的父亲节点,但是如果该节点是右孩子就比较复杂了,需要找到该节点父亲节点是左孩子的节点,举个例子,节点 7 是 4 的右孩子,那么看 4 是不是 4 的父亲节点的左孩子,不是,继续向上找,2 是 父亲节点 1 的左孩子,结束循环,说明 7 的下一个节点是 2 的父亲节点 3(这里有点小饶,一缕就清楚)
public TreeNode nextNode(TreeNode node) {
TreeNode pNext = null;
TreeNode parent = searchParent(root, node.getData());//找到该节点的父亲节点
if (node.getRight() != null) {//如果存在右孩子,找右孩子的最左孩子就是下一个节点
TreeNode temp = node.getRight();
while (temp.getLeft() != null) {
temp = temp.getLeft();
}
pNext = temp;
// 如果不存在父亲节点,说明是根节点且没有右孩子,说明已经是最后一个节点了
} else if (parent != null) {// 如果存在父亲节点,判断该节点是左孩子还有右孩子,左孩子直接返回父亲节点,右孩子需要单独判断
TreeNode temp = node;
TreeNode pParent = parent;
if (pParent.getRight() != null && pParent.getRight().getData() == temp.getData()) {//是右孩子
while (pParent != null && temp == pParent.getRight()) {
//向上找到第一个是父亲节点是左孩子的情况
temp = pParent;
pParent = searchParent(root, pParent.getData());
}
pNext = pParent;
}
return pParent;
}
return pNext;
}