二叉搜索树

import java.util.ArrayList;
import java.util.List;


public class BinarySearchTree<Key extends Comparable<Key>, Value> {

    Node<Key, Value> root;

    Node get(Node node, Key k) {
        if (node == null) {
            return null;
        }
        int cmp = node.k.compareTo(k);
        if (cmp > 0) {
            return get(node.left, k);
        } else if (cmp < 0) {
            return get(node.right, k);
        } else {
            return node;
        }
    }

    int size(Node node) {
        if (node == null) {
            return 0;
        }
        return node.N;
    }

    void put(Key k, Value v) {
        root = put(root, k, v);
    }

    Node put(Node node, Key k, Value v) {
        if (node == null) {
            return new Node(k, v, 1);
        }
        int cmp = node.k.compareTo(k);
        if (cmp > 0) {
            node.left = put(node.left, k, v);
        } else if (cmp < 0) {
            node.right = put(node.right, k, v);
        } else {
            node.v = v;
        }
        node.N = size(node.left) + size(node.right) + 1;
        return node;
    }

    void HierarchicalTraversal() {
        HierarchicalTraversal(root);
    }

    void deleteMin() {
        root = deleteMin(root);
    }

    Node deleteMin(Node node) {
        if (node == null) {
            return null;
        }
        if (node.left == null) {
            return node.right;
        } else {
            node.N = node.N - 1;
            node.left = deleteMin(node.left);
            return node;
        }
    }

    /**
     * 层序遍历
     */
    void HierarchicalTraversal(Node node) {
        if (node == null) {
            return;
        }
        List<Node> list = new ArrayList<Node>();
        list.add(node);
        while (!list.isEmpty()) {
            Node tmp = list.remove(0);
            System.out.println(tmp.k + "--" + tmp.v + ",N=" + tmp.N);
            if (tmp.left != null) {
                list.add(tmp.left);
            }
            if (tmp.right != null) {
                list.add(tmp.right);
            }
        }
    }

    public static void main(String[] args) {
        Integer[] keys = {6, 5, 4, 2, 3};
        char[] values = {'a', 'b', 'c', 'd', 'e'};
        BinarySearchTree<Integer, Character> bst = new BinarySearchTree<Integer, Character>();
        for (int i = 0; i < keys.length; i++) {
            bst.put(keys[i], values[i]);
        }

        bst.HierarchicalTraversal();
        bst.deleteMin();
        System.out.println("----------------");
        bst.HierarchicalTraversal();
        bst.deleteMin();
        System.out.println("----------------");
        bst.HierarchicalTraversal();
        bst.deleteMin();
        System.out.println("----------------");
        bst.HierarchicalTraversal();
    }
}

class Node<Key extends Comparable<Key>, Value> {
    Node left, right;
    /**
     * N = 该节点及其后代的数量
     */
    int N;
    Key k;
    Value v;

    public Node(Key k, Value v, int N) {
        this.k = k;
        this.v = v;
        this.N = N;
    }
}

参考书籍:《算法》[美] Robert Sedgewick,[美] Kevin Wayne 著,谢路云 译

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值