详谈二叉搜索树

详谈二分搜索树

在这里插入图片描述

二分搜索树的概念

在了解二分搜索树之前,我们得弄清楚什么是二叉树。

什么是二叉树?

在这里插入图片描述

二分搜索树

在这里插入图片描述

二分搜索树的创建以及实现其方法

1.内部结构

内部有一个结点类,作为二分搜索树的成员变量使用

private class Node{
        public E e;//结点元素
        public Node left,right;//左右结点
        public Node (E e){//初始化
            this.e = e;
            left = null;
            right = null;
        }
        public  String toString(){
            return e.toString();
        }
    }
2.构造方法与成员变量

初始化成员变量

    private Node root;//根节点
    private int size;//有效结点个数
    public BinarySearchTree(){//构造方法
        size =0;
        root = null;
    }
3.增添元素

根据元素值大小决定去向,小于父亲节点在左边,大于在右边

 public int size(){//获取有效长度
        return size;
    }
    public boolean isEmpty(){//判断是否为空
        return size==0&&root==null;
    }
    public void clear(){
        size =0;
        root = null;
    }
    public void add(E e){增添元素
        root = add(root,e);//在根节点的基础上去添加
    }
    private Node add(Node node,E e){
        if(node==null){
            size++;
            return  new Node(e);
        }
        if(e.compareTo(node.e)>0){//e大于父节点的值
            node.right = add(node.right,e);
        }else if(e.compareTo(node.e)<0){
            node.left = add(node.left,e);
        }else {
            node.e = e;
        }
        return node;
    }
4.遍历元素

遍历元素总共有四种种先序、后序、中序和层次遍历。

private void preOrderNR(){//先序非递归遍历
        Stack<Node> stack = new Stack<>();
        stack.push(root);
        while(!stack.isEmpty()){
            Node node = stack.pop();
            System.out.print(node.e+"  ");
            if (node.right!=null){//利用栈先入右节点,在入左节点  先进后出 后进先出
                stack.push(node.right);
            } if(node.left!=null){
                stack.push(node.left);
            }

        }

    }
    private void preOrder(Node node){//先序递归遍历
        if(node == null){
            return;
        }
        System.out.print(node.e+"  ");
        preOrder(node.left);
        preOrder(node.right);
    }
    public void inOrder(){
            inOrder(root);
        System.out.println("=========");
            inOrderNR();
    }
    public void inOrder(Node node){//中序递归遍历
        if(node==null){
            return;
        }
        inOrder(node.left);
        System.out.print(node.e+"  ");
        inOrder(node.right);
    }
    public void inOrderNR(){//中序非递归遍历
        Stack<Node> stack = new Stack();
        Node p =root;
        while(p!=null){
            stack.push(p);
            p = p.left;
        }
        while(!stack.isEmpty()){
            Node ret = stack.pop();
            System.out.print(ret.e+"  ");
            if(ret.right!=null){
                p = ret.right;
                while(p!=null){
                    stack.push(p);
                    p = p.left;
                }
            }
        }
    }
    public void postOrder(){
        postOrder(root);
        System.out.println();
        postOrderNR();
    }
     private void postOrder(Node node){//后序递归遍历
        if(node == null){
            return ;
        }
        postOrder(node.left);
        postOrder(node.right);
        System.out.print(node.e+"  ");
    }
    private void postOrderNR() {//后序非递归遍历
        Stack<Node> stack = new Stack<>();
        Node node = root;
        Node pre = null;
        while(!stack.isEmpty()||node!=null){
            while (node!=null){
                stack.push(node);
                node = node.left;
            }
            if (!stack.isEmpty()){
                node = stack.pop();
                if(node.right==null||pre==node.right){
                    System.out.print(node.e+"  ");
                    pre = node;
                    node =null;
                }else{
                    stack.push(node);
                    node = node.right;
                }
            }

        }
    }
    public void levelOrder(){//层序遍历
         Queue<Node> que = new LinkedList<>();
         que.add(root);
         while(!que.isEmpty()){
             Node e = que.poll();
             System.out.print(e.e);
             if(e.left!=null){
                 que.add(e.left);
             }
             if(e.right!=null){
                 que.add(e.right);
             }
         }
    }
5.判断元素是否存在

目的是为了判断一个元素是否在二分查找树中

 public boolean contains(E e){

        return contains(root,e);
    }
    private boolean contains(Node node,E e){
        if(node == null){
            return false;
        }
        if(e.compareTo(node.e)==0){
            return true;
        }
        return e.compareTo(node.e)>0 ? contains(node.right,e):contains(node.left,e);
    }
6.remove移除一个结点

删除一个结点及元素值

public void remove(E e){
        root = remove(root,e);
    }
    public Node remove(Node node,E e){
        if(node==null){
            return null;
        }
        if(e.compareTo(node.e)>0){
            node.right = remove(node.right,e);
        }else if(e.compareTo(node.e)<0){
            node.left = remove(node.left,e);
        }else{
            if(node.left==null){
                Node rightNode = node.right;
                node.right = null;
                size--;
                return  rightNode;
            }
            if(node.right==null){
                Node leftNode = node.left;
                node.left = null;
                size--;
                return leftNode;
            }
            Node successor = minimun(minimun(node.right));
            successor.right = removemin(node.right);
            successor.left = node.left;
            node.left =null;
            node.right = null;
            size--;
            return  successor;
        }
        return null;
    }

最终整体代码

package com.数据结构.xiaoniu.;

import com.sun.org.apache.xalan.internal.xsltc.compiler.util.StringStack;

import java.util.Iterator;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Stack;

public class BinarySearchTree<E extends Comparable<E>> implements Iterable{
    private class Node{
        public E e;
        public Node left,right;
        public Node (E e){
            this.e = e;
            left = null;
            right = null;
        }
        public  String toString(){
            return e.toString();
        }
    }
    private Node root;
    private int size;
    public BinarySearchTree(){
        size =0;
        root = null;
    }
    public int size(){
        return size;
    }
    public boolean isEmpty(){
        return size==0&&root==null;
    }
    public void clear(){
        size =0;
        root = null;
    }
    public void add(E e){

        root = add(root,e);
    }
    private Node add(Node node,E e){
        if(node==null){
            size++;
            return  new Node(e);
        }
        if(e.compareTo(node.e)>0){
            node.right = add(node.right,e);
        }else if(e.compareTo(node.e)<0){
            node.left = add(node.left,e);
        }else {
            node.e = e;
        }
        return node;
    }
    public boolean contains(E e){

        return contains(root,e);
    }
    private boolean contains(Node node,E e){
        if(node == null){
            return false;
        }
        if(e.compareTo(node.e)==0){
            return true;
        }
        return e.compareTo(node.e)>0 ? contains(node.right,e):contains(node.left,e);
    }
    public void preOrder(){
       preOrder(root);
        System.out.println();
       preOrderNR();

    }
    private void preOrderNR(){
        Stack<Node> stack = new Stack<>();
        stack.push(root);
        while(!stack.isEmpty()){
            Node node = stack.pop();
            System.out.print(node.e+"  ");
            if (node.right!=null){//利用栈先入右节点,在入左节点  先进后出 后进先出
                stack.push(node.right);
            } if(node.left!=null){
                stack.push(node.left);
            }

        }

    }
    private void preOrder(Node node){
        if(node == null){
            return;
        }
        System.out.print(node.e+"  ");
        preOrder(node.left);
        preOrder(node.right);
    }
    public void inOrder(){
            inOrder(root);
        System.out.println("=========");
            inOrderNR();
    }
    public void inOrder(Node node){
        if(node==null){
            return;
        }
        inOrder(node.left);
        System.out.print(node.e+"  ");
        inOrder(node.right);
    }
    public void inOrderNR(){
        Stack<Node> stack = new Stack();
        Node p =root;
        while(p!=null){
            stack.push(p);
            p = p.left;
        }
        while(!stack.isEmpty()){
            Node ret = stack.pop();
            System.out.print(ret.e+"  ");
            if(ret.right!=null){
                p = ret.right;
                while(p!=null){
                    stack.push(p);
                    p = p.left;
                }
            }
        }
    }
    public void postOrder(){
        postOrder(root);
        System.out.println();
        postOrderNR();
    }

    private void postOrderNR() {
        Stack<Node> stack = new Stack<>();
        Node node = root;
        Node pre = null;
        while(!stack.isEmpty()||node!=null){
            while (node!=null){
                stack.push(node);
                node = node.left;
            }
            if (!stack.isEmpty()){
                node = stack.pop();
                if(node.right==null||pre==node.right){
                    System.out.print(node.e+"  ");
                    pre = node;
                    node =null;
                }else{
                    stack.push(node);
                    node = node.right;
                }
            }

        }
    }
    public void levelOrder(){//层序遍历
         Queue<Node> que = new LinkedList<>();
         que.add(root);
         while(!que.isEmpty()){
             Node e = que.poll();
             System.out.print(e.e);
             if(e.left!=null){
                 que.add(e.left);
             }
             if(e.right!=null){
                 que.add(e.right);
             }
         }
    }
    public E minimum(){
        if(size==0){
            throw new IllegalArgumentException("Is NULL!");
        }
        Node n = root;
        while (n.left!=null){
            n=n.left;
        }
        return n.e;
    }
    public E maxmun(){
        if(size==0){
            throw new IllegalArgumentException("Is NULL!");
        }
        Node n = root;
        while (n.right!=null){
            n=n.right;
        }
        return n.e;
    }
    public E removeMin(){//删除最小二叉树中的节点
        if(size==0){
            throw new IllegalArgumentException("Is NULL!");
        }
        Node n = root;
        while (n.left.left!=null){
            n=n.left;
        }
        E ret = n.left.e;
        n.left = n.left.right;
        return ret;
    }
    public E removeMax(){
        if(size==0){
            throw new IllegalArgumentException("Is NULL!");
        }
        Node n = root;
        while (n.right.right!=null){
            n=n.right;
        }
        E ret = n.right.e;
        n.right = n.right.left;
        return ret;
    }
    public E removemin(){
        E ret = minimum();
        root = removemin(root);
        return ret;
    }
    public E removemax(){
        E ret = maxmun();
        root = removemax(root);
        return ret;
    }
    private Node removemax(Node node){
        if(node.right==null){
            Node leftNode = node.left;
            node.left = null;
            size--;
            return leftNode;
        }
        node = node.right;
        node.right = removemax(node.right);
        return node;
    }
    public Node removemin(Node node){
        if(node.left==null){
            Node rightNode = node.right;
            node.right = null;
            size--;
            return rightNode;
        }
        node.left = removemin(node.left);
        return node;
    }

    public Node minimun(Node node){
        if(node.left==null){
            return node;
        }
        return minimun(node.left);
    }
    public Node maximum(Node node){
        if(node.right==null){
            return node;
        }
        return minimun(node.right);
    }
    public void remove(E e){
        root = remove(root,e);
    }
    public Node remove(Node node,E e){
        if(node==null){
            return null;
        }
        if(e.compareTo(node.e)>0){
            node.right = remove(node.right,e);
        }else if(e.compareTo(node.e)<0){
            node.left = remove(node.left,e);
        }else{
            if(node.left==null){
                Node rightNode = node.right;
                node.right = null;
                size--;
                return  rightNode;
            }
            if(node.right==null){
                Node leftNode = node.left;
                node.left = null;
                size--;
                return leftNode;
            }
            Node successor = minimun(minimun(node.right));
            successor.right = removemin(node.right);
            successor.left = node.left;
            node.left =null;
            node.right = null;
            size--;
            return  successor;
        }
        return null;
    }
    private void postOrder(Node node){
        if(node == null){
            return ;
        }
        postOrder(node.left);
        postOrder(node.right);
        System.out.print(node.e+"  ");
    }
    public String toString(){
        StringBuilder res =new StringBuilder();
        gennerateBSTString(root,0,res);
        return res.toString();
    }

    private void gennerateBSTString(Node node, int depth, StringBuilder res) {
        if(node==null){
            res.append(generateDepthString(depth)+node.e+"\n");
            return;
        }
        res.append(generateDepthString(depth)+node.e+"\n");
        gennerateBSTString(node.left,depth+1,res);
        gennerateBSTString(node.right,depth+1,res);
    }

    private String generateDepthString(int depth) {
        StringBuilder sb =new StringBuilder();
        for(int i=0;i<depth;i++){
            sb.append("--");
        }
        return  sb.toString();
    }

    @Override
    public Iterator<E> iterator() {
        return new BinarySearchTreeIterator();
    }
    private class BinarySearchTreeIterator implements Iterator<E>{
        private LinkedList<Node> list = new LinkedList<>();
        public BinarySearchTreeIterator(){
            Stack<Node> stack = new Stack<>();
            Node p = root;
            while(p!=null){
                stack.push(p);
                p =p.left;
            }
            while (!stack.isEmpty()){
                Node n = stack.pop();
                list.add(n);
                if(n.right!=null){
                    p = n.right;
                    while(p!=null){
                        stack.push(p);
                        p = p.left;
                    }
                }
            }
        }
        @Override
        public boolean hasNext() {
            return !list.isEmpty();
        }

        @Override
        public E next() {

            return (E) list.removeFirst();
        }
    }
}

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

牛牛最爱喝兽奶

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值