【算法】求一个节点的后继节点(中序遍历)

public class Node<V> {
    public V value;
    public Node left; //左孩子
    public Node right; //右孩子
    public Node parent; //父节点

    public Node(V value) {
        this.value = value;
    }
}

以上树节点的结构中增加了指向父节点的引用。

基于上述节点结构,若想求中序遍历中一个节点的后继节点,该如何求?

在中序遍历中,对于一个节点来说,若它有右子树,则右子树上最左节点即为该节点的后继节点。若无右子树,则从该节点开始,沿着父节点引用往上扫描,看是否在父节点的左分支上,若在,则该父节点即为后继节点,若一直扫描到父节点为null,则这个节点是树的最右节点,无后继节点。

public class SuccessorNode {

    public static Node getSuccessorNode(Node node) {
        if (node == null) {
            return null;
        }
        if (node.right != null) { //右子树存在则右子树上最左节点即为后继节点
            return getLeftMost(node.right);
        } else { //无右子树
            Node parent = node.parent;
            while (parent != null && parent.left != node) {//父不空且该节点不是其父的左孩子
                node = parent;
                parent = node.parent;
            }
            return parent;
        }
    }
    //求以node为根节点的树的最左节点。
    public static Node getLeftMost(Node node) {
        if (node == null) {
            return null;
        }
        while (node.left != null) {
            node = node.left;
        }
        return node;
    }
}
posted @ 2022-01-23 15:09  hzyuan  阅读(115)  评论(0)    收藏  举报