| 
 | 1 | +import java.util.LinkedList;  | 
 | 2 | +import java.util.Objects;  | 
 | 3 | +import java.util.Queue;  | 
 | 4 | + | 
 | 5 | +public class DesignHashMap {  | 
 | 6 | +    class MyHashMap {  | 
 | 7 | + | 
 | 8 | +        private final int initialSize = 100;  | 
 | 9 | +        private final Bucket[] buckets = new Bucket[initialSize];  | 
 | 10 | + | 
 | 11 | +        public MyHashMap() {  | 
 | 12 | +            for (int index = 0 ; index < buckets.length ; index++) {  | 
 | 13 | +                buckets[index] = new Bucket();  | 
 | 14 | +            }  | 
 | 15 | +        }  | 
 | 16 | + | 
 | 17 | +        public void put(int key, int value) {  | 
 | 18 | +            int hashCode = key % buckets.length;  | 
 | 19 | +            buckets[hashCode].put(key, value);  | 
 | 20 | +        }  | 
 | 21 | + | 
 | 22 | +        public int get(int key) {  | 
 | 23 | +            return buckets[key % buckets.length].get(key);  | 
 | 24 | +        }  | 
 | 25 | + | 
 | 26 | +        public void remove(int key) {  | 
 | 27 | +            buckets[key % buckets.length].remove(key);  | 
 | 28 | +        }  | 
 | 29 | + | 
 | 30 | +        private class Bucket {  | 
 | 31 | +            Queue<Pair> list = new LinkedList<>();  | 
 | 32 | + | 
 | 33 | + | 
 | 34 | +            public boolean contains(int element) {  | 
 | 35 | +                return list.stream().anyMatch(pair -> pair.key == element);  | 
 | 36 | +            }  | 
 | 37 | + | 
 | 38 | +            public void put(int key, int value) {  | 
 | 39 | +                if (contains(key)) {  | 
 | 40 | +                    Pair match = list.stream().filter(pair -> pair.key == key).findFirst().get();  | 
 | 41 | +                    match.value = value;  | 
 | 42 | +                } else {  | 
 | 43 | +                    list.add(new Pair(key, value));  | 
 | 44 | +                }  | 
 | 45 | +            }  | 
 | 46 | + | 
 | 47 | +            public int get(int key) {  | 
 | 48 | +                return contains(key) ? list.stream().filter(pair -> pair.key == key).findFirst().get().value : -1;  | 
 | 49 | +            }  | 
 | 50 | + | 
 | 51 | +            public void remove(int key) {  | 
 | 52 | +                if (contains(key)) {  | 
 | 53 | +                    list.remove(new Pair(key, get(key)));  | 
 | 54 | +                }  | 
 | 55 | +            }  | 
 | 56 | +        }  | 
 | 57 | + | 
 | 58 | +        private class Pair {  | 
 | 59 | +            private final int key;  | 
 | 60 | +            private int value;  | 
 | 61 | + | 
 | 62 | +            Pair(int key, int value) {  | 
 | 63 | +                this.key = key;  | 
 | 64 | +                this.value = value;  | 
 | 65 | +            }  | 
 | 66 | + | 
 | 67 | +            @Override  | 
 | 68 | +            public boolean equals(Object o) {  | 
 | 69 | +                if (this == o) return true;  | 
 | 70 | +                if (o == null || getClass() != o.getClass()) return false;  | 
 | 71 | +                Pair pair = (Pair) o;  | 
 | 72 | +                return key == pair.key && value == pair.value;  | 
 | 73 | +            }  | 
 | 74 | + | 
 | 75 | +            @Override  | 
 | 76 | +            public int hashCode() {  | 
 | 77 | +                return Objects.hash(key, value);  | 
 | 78 | +            }  | 
 | 79 | +        }  | 
 | 80 | +    }  | 
 | 81 | +}  | 
0 commit comments