系统设计之HashSet

最新推荐文章于 2024-07-04 11:18:33 发布

zjshuster

最新推荐文章于 2024-07-04 11:18:33 发布

阅读量110

点赞数 1

分类专栏：算法职场总结文章标签：数据结构哈希算法 java

本文链接：https://blog.csdn.net/zjshuster/article/details/125395274

版权

算法同时被 2 个专栏收录

52 篇文章 0 订阅

订阅专栏

职场总结

20 篇文章 0 订阅

订阅专栏

该博客介绍了一个简单的Java实现，名为MyHashSet，它基于数组和链表的模式，模仿HashMap的功能但不存储值。MyHashSet包含了添加、删除和检查元素是否存在的方法，通过计算哈希值来定位元素。内部使用了链表处理哈希冲突。

摘要由CSDN通过智能技术生成

参考上一篇hashmap设计思路hashmap设计，这里去除值的属性，采用数组+链表的模式实现hashSet的实现

public class MyHashSet {

    final ListNode[] nodes;

    public MyHashSet() {
        this.nodes = new ListNode[124981];
    }

    public void add(int key) {
        int index = idx(key);

        if (nodes[index] == null) {
            nodes[index] = new ListNode(key);
        } else {
            ListNode node = nodes[index];
            ListNode pre = null;
            while (node != null && node.key != key) {
                pre = node;
                node = node.next;

            }
            if (node == null) {
                pre.next = new ListNode(key);
            }
        }
    }

    private int idx(int key) {
        return Integer.hashCode(key) % nodes.length;
    }

    public void remove(int key) {
        int index = idx(key);
        if (nodes[index] != null) {
            ListNode node = nodes[index];
            ListNode pre = null;
            while (node != null && node.key != key) {
                pre = node;
                node = node.next;
            }
            if (node != null ) {
                if(pre != null)
                pre.next = node.next;
                else nodes[index] = null;
            }
        }
    }

    public boolean contains(int key) {
        int index = idx(key);
        if (nodes[index] == null) {
            return false;
        } else {
            ListNode node = nodes[index];
            while (node != null && node.key != key) {
                node = node.next;
            }
            return node != null;
        }
    }

    class ListNode {
        int key;
        ListNode next;

        public ListNode(int key) {
            this.key = key;
        }

    }
}