完全二叉树-添加节点(Java实现)

算法思路
  1. 找出最后一个节点的父节点
  2. 如果该父节点没有右子节点,则新节点就添加为它的右子节点
  3. 如果该父节点有右子节点,则新节点添加为该父节点下一个节点的左子节点
  4. 只有该父节点是根节点的情况下才可能出现没有左子节点的情况,此时新节点添加到根节点左子节点

算法实现

二叉树的实现
  1. 节点类
    • 左右子节点的引用Node<T> left Node<T> right
    • 保存的数据T data
    • 重写toString()方便查看输出结果
class Node<T> {
    T data;
    Node<T> left = null;
    Node<T> right = null;
    public Node(T data) {
        this.data = data;
    }

    @Override
    public String toString() {
        return "" + data;
    }
}
  1. 二叉树类
    • 根节点Node<T> root
    • 判断是否为空的方法isEmpty()
    • List<Node<T>> nodes模拟数组储存节点数据,方便索引找到节点
    • breadthFirstTravel()广度优先遍历方法,方便检查是否添加成功
    • 关于广度优先遍历查看 https://www.jianshu.com/p/4a346dffa2e5
public class CompleteBinaryTree<T> {
    Node<T> root;
    // nodes集合按添加顺序储存树的节点
    List<Node<T>> nodes = new ArrayList<>();
     public boolean isEmpty() {
        return root == null;
    }
    public void breadthFirstTravel() {
        if (isEmpty()) {
            System.out.println("树为空");
            return;
        }
        Queue<Node<T>> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            Node<T> cur = queue.remove();
            System.out.println(cur.data);
            if (cur.left != null) {
                queue.add(cur.left);
            }
            if (cur.right != null) {
                queue.add(cur.right);
            }
        }
    }
}
具体步骤
  1. 判断树是否为空,是则根节点赋值为该新节点并结束方法
  2. 得到最后一个节点的父节点索引int index = (nodes.size() - 1 - 1) / 2;
    • nodes.size()是节点数组长度
    • 长度-1是最后一个节点的索引
    • (节点索引 - 1) / 2是该节点的父节点索引
  3. 如果父节点有右子节点nodes.get(index).right != null
    则新节点添加到该父节点下一节点的左子节点nodes.get(index + 1).left = node
  4. 如果父节点有左子节点nodes.get(index).left != null
    则新节点添加到该父节点的右子节点nodes.get(index).right = node
  5. 特殊情况就是树只有根节点(上面已经分析过),直接添加到根节点左子节点nodes.get(index).left = node
    /**
     * 添加节点
     *
     * @param data 节点储存数据
     * @return void
     */
    public void add(T data) {
        Node<T> node = new Node<>(data);
        nodes.add(node);
        if (isEmpty()) {  // 树空时添加根节点
            root = node;
            return;
        }
        // 得到最后一个节点的父节点索引
        int index = (nodes.size() - 1 - 1) / 2;
        // 如果该父节点有右子节点
        if (nodes.get(index).right != null) {
            // 则新节点的父节点是当前父节点在广度优先上的下一个节点
            nodes.get(index + 1).left = node;
            // 如果该父节点有左子节点(并且没有右子节点)
        } else if (nodes.get(index).left != null) {
            // 则新节点是该父节点的右子节点
            nodes.get(index).right = node;
            // 最后一种情况: 该父节点没有左子节点
        } else {
            nodes.get(index).left = node;
        }
    }
测试
  • 循环添加1-9保存到树节点中
  • 广度优先遍历
    public static void main(String[] args) {
        CompleteBinaryTree<Integer> tree = new CompleteBinaryTree<>();
        for (int i = 1; i < 10; i++) {
            tree.add(i);
        }
        tree.breadthFirstTravel();
    }

输出结果:1 2 3 4 5 6 7 8 9

完整代码
class Node<T> {
    T data;
    Node<T> left = null;
    Node<T> right = null;
    public Node(T data) {
        this.data = data;
    }

    @Override
    public String toString() {
        return "" + data;
    }
}
public class CompleteBinaryTree<T> {
    Node<T> root;
    // nodes集合按添加顺序储存树的节点
    List<Node<T>> nodes = new ArrayList<>();
     public boolean isEmpty() {
        return root == null;
    }
    public void breadthFirstTravel() {
        if (isEmpty()) {
            System.out.println("树为空");
            return;
        }
        Queue<Node<T>> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            Node<T> cur = queue.remove();
            System.out.println(cur.data);
            if (cur.left != null) {
                queue.add(cur.left);
            }
            if (cur.right != null) {
                queue.add(cur.right);
            }
        }
    }
    /**
     * 添加节点
     *
     * @param data 节点储存数据
     * @return void
     */
    public void add(T data) {
        Node<T> node = new Node<>(data);
        nodes.add(node);
        if (isEmpty()) {  // 树空时添加根节点
            root = node;
            return;
        }
        // 得到最后一个节点的父节点索引
        int index = (nodes.size() - 1 - 1) / 2;
        // 如果该父节点有右子节点
        if (nodes.get(index).right != null) {
            // 则新节点的父节点是当前父节点在广度优先上的下一个节点
            nodes.get(index + 1).left = node;
            // 如果该父节点有左子节点(并且没有右子节点)
        } else if (nodes.get(index).left != null) {
            // 则新节点是该父节点的右子节点
            nodes.get(index).right = node;
            // 最后一种情况: 该父节点没有左子节点
        } else {
            nodes.get(index).left = node;
        }
    }
    public static void main(String[] args) {
        CompleteBinaryTree<Integer> tree = new CompleteBinaryTree<>();
        for (int i = 1; i < 10; i++) {
            tree.add(i);
        }
        tree.breadthFirstTravel();
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。