基于数组-实现简易版HashMap集合

太过爱你忘了你带给我的痛 2023-10-15 06:52 95阅读 0赞

#### 基于数组-实现简易版HashMap集合 ####

import java.util.ArrayList;
    import java.util.List;
    
    //基于数组，实现简易版HashMap集合。
    
    public class SimpleHashMap<K, V> {
        //    初始容量
        private static final int DEFAULT_CAPACITY = 16;
        private List<Entry<K, V>>[] buckets;
    
        public SimpleHashMap() {
            buckets = new List[DEFAULT_CAPACITY];
            for (int i = 0; i < DEFAULT_CAPACITY; i++) {
                buckets[i] = new ArrayList<>();
            }
        }
    
        // 向HashMap中插入键值对
        public void put(K key, V value) {
            int index = hash(key);  // 计算哈希值得到索引
            List<Entry<K, V>> bucket = buckets[index];  // 获取对应索引的桶
    
            // 遍历桶中的键值对，如果键已存在，则更新值
            for (Entry<K, V> entry : bucket) {
                if (entry.getKey().equals(key)) {
                    entry.setValue(value);
                    return;
                }
            }
    
            // 如果键不存在，则将键值对添加到桶中
            bucket.add(new Entry<>(key, value));
        }
    
        // 根据键获取值
        public V get(K key) {
            int index = hash(key);  // 计算哈希值得到索引
            List<Entry<K, V>> bucket = buckets[index];  // 获取对应索引的桶
    
            // 遍历桶中的键值对，查找并返回匹配的值
            for (Entry<K, V> entry : bucket) {
                if (entry.getKey().equals(key)) {
                    return entry.getValue();
                }
            }
    
            // 如果键不存在，则返回null
            return null;
        }
    
        // 计算哈希值并获取对应的索引
        private int hash(K key) {
            return key.hashCode() % DEFAULT_CAPACITY;
        }
    
        // 嵌套的Entry类，用于存储键值对
        private static class Entry<K, V> {
            private K key;
            private V value;
    
            public Entry(K key, V value) {
                this.key = key;
                this.value = value;
            }
    
            public K getKey() {
                return key;
            }
    
            public V getValue() {
                return value;
            }
    
            public void setValue(V value) {
                this.value = value;
            }
        }
    
        // 测试示例
        public static void main(String[] args) {
            SimpleHashMap<String, Integer> map = new SimpleHashMap<>();
            map.put("one", 1);
            map.put("two", 2);
            map.put("three", 3);
    
            System.out.println(map.get("one"));    // Output: 1
            System.out.println(map.get("two"));    // Output: 2
            System.out.println(map.get("three"));  // Output: 3
            System.out.println(map.get("four"));   // Output: null
        }
    
    }