大家好,都吃晚饭了吗?我是Kaiqisan,是一个已经走出社恐的一般生徒,最近的问题都围绕着树齁
问题
现有二叉树,每个节点多一个指针,会指向父节点
现在要找目标节点的后继节点,(中序遍历中的当前节点的下一个节点)
进阶要求:不要遍历整棵树
参考树类
class Tree<T>{
public T val;
public Tree left = null;
public Tree right = null;
public Tree parent;
public Tree(T val) {
this.val = val;
}
}
思路
好好利用上面的核心机制— 中序遍历— 左中右的顺序
然后就开始了
先把自己当做中间节点,查看是否有右孩子,如果有就开始遍历,找右孩子的最左节点(无脑遍历左节点)
如果没有右节点的话就要去找它的父节点(假定自己是是左,要找中),利用父节点查看自己是否为左孩子,如果是左孩子的话就输出自己的父节点,如果是右孩子的话就接着往父节点遍历,直到找到某个节点是其父节点的左孩子,如果实在找不到(此时已经指针已经摸到了根节点)就表示自己是中序遍历的最后一个节点(此时已经证实自己是右),此时返回null
代码
static TreeHasParent searchNextNode(TreeHasParent head) {
if (head.right != null) {
head = head.right;
while (head.left != null) {
head = head.left;
}
return head;
} else {
while (head.parent != null && head.parent.right == head) {
head = head.parent;
}
return head.parent == null ? null : head.parent;
}
}
总结
没有总结,基础内容,好好掌握。