Java开发 - 树（二叉树，二叉排序树，红黑树）（二）

判断root是否为null
  -为null,则将元素封装成节点,称为root,返回true
  -不为null,和root进行比较,比较的目的是将元素尝试添加为root的left/right子树
    - 和root相等,添加失败,返回false,结束
    - 大于>root,尝试添加为root的right;判断right是否为null,
      - 为null,则让新元素封装成节点,称为right,添加成功,返回true
      - 不为null,继续和right的节点进行比较,尝试将元素添加为right的left/right子树
    - 小于,同理

//添加元素
    public boolean add(E e){
        //判断root是否为null，为null，成为根节点
        if (root==null) {
            root = new Node(e);
            return true;
        }
        //root不为null，添加
        return root.append(e);
    }

       //向某个节点上添加给定的元素,尝试让新元素称为其左/右子树
        public boolean append(E e) {
            if (e.compareTo(ele) == 0) {
                return false;
            } else if (e.compareTo(ele) < 0) {
                if (left == null) {
                    left = new Node(e);
                    return true;
                }
                return left.append(e);
            } else {
                if (right == null) {
                    right = new Node(e);
                    return true;
                }
                return right.append(e);
            }
        }

    //根据元素查询对应的节点对象
    public Node get(E e){
        //判断root是否为null
        if (root==null)
            return null;
        //root不为空,则判断root是否为目标节点
        return root.isDestNode(e);
    }

        //用于判断调用方法的节点是否为目标节点
        public Node isDestNode(E e) {
            //判断e是否和当前节点的元素相等,若相等,说明为目标节点,返回当前节点即可
            if (e.compareTo(ele)==0)
                return this;
            else if (e.compareTo(ele)>0){  //e大于当前节点,到right上继续查找
                //若right为null,说明目标元素不存在树中,返回null
                if (right==null)
                    return null;
                return right.isDestNode(e);
            }else {  //e小于当前节点,到left上继续查找
                if (left==null)
                    return null;
                return left.isDestNode(e);
            }
        }

set(E ele, E newEle)

    //重写toString
    @Override
    public String toString() {
        //判断root是否为null,为null,返回"[]"
        if (root==null)
            return "[]";
        //进行中序遍历
        StringBuilder builder = new StringBuilder("[");
        return root.middleOrder(builder).replace(builder.length()-1,builder.length(),"]").toString();

        public StringBuilder middleOrder(StringBuilder builder) {
            //左
            if (left!=null){
                left.middleOrder(builder);
            }
            //取根的值
            builder.append(ele).append(",");
            //取右
            if (right!=null){
                right.middleOrder(builder);
            }
            //以上三步操作结束,得到遍历树后的元素拼接结果，将最后的逗号替换为]
            return builder;
        }