手写一个简单的HashMap

/**
 * @author 曹见朋
 * @create 2024-09-20-8:17
 */
public class MyHashMap {
    private static class Node {
        private Integer key;
        private Integer value;

        Node(Integer key, Integer value) {
            this.key = key;
            this.value = value;
        }
    }

    private Integer capacity; // 容量
    private Integer size; // 元素个数
    private LinkedList<Node>[] table; // 桶数组

    private static final Integer DEFAULT_CAPACITY = 16;


    public MyHashMap() {
        this.capacity = DEFAULT_CAPACITY;
        this.size = 0;
        this.table = new LinkedList[DEFAULT_CAPACITY];
    }

    public MyHashMap(Integer capacity) {
        this.capacity = capacity;
        this.size = 0;
        this.table = new LinkedList[capacity];
    }

    private Integer getIndex(Integer key) {
        return (capacity - 1) & ((key.hashCode() ^ (key.hashCode() >>> 16)));
    }

    public void put(Integer key, Integer value) {
        int index = getIndex(key);
        if (table[index] == null) {
            table[index] = new LinkedList<>();
        }
        for (Node node : table[index]) {
            if (node.key.equals(key)) {
                node.value = value;
                return;
            }
        }
        table[index].addLast(new Node(key, value));
        size++;
    }

    public Integer get(Integer key) {
        int index = getIndex(key);
        if (table[index] == null) {
            throw new RuntimeException("元素不存在");
        }
        for (Node node : table[index]) {
            if (node.key.equals(key)) {
                return node.value;
            }
        }
        return null;
    }

    public void remove(Integer key){
        int index = getIndex(key);
        if(table[index] == null){
            throw new RuntimeException("素不存在");
        }
        for(Node node : table[index]) {
            if (node.key.equals(key)) {
                table[index].remove(node);
                size--;
            }
        }
    }

    public Integer size(){
        return size;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值