Java实现 LeetCode 706 设计哈希映射(数组+链表)

706. 设计哈希映射

不使用任何内建的哈希表库设计一个哈希映射

具体地说,你的设计应该包含以下的功能

put(key, value):向哈希映射中插入(,)的数值对。如果键对应的值已经存在,更新这个值。
get(key):返回给定的键所对应的值,如果映射中不包含这个键,返回-1remove(key):如果映射中存在这个键,删除这个数值对。

示例:

MyHashMap hashMap = new MyHashMap();
hashMap.put(1, 1);          
hashMap.put(2, 2);         
hashMap.get(1);            // 返回 1
hashMap.get(3);            // 返回 -1 (未找到)
hashMap.put(2, 1);         // 更新已有的值
hashMap.get(2);            // 返回 1 
hashMap.remove(2);         // 删除键为2的数据
hashMap.get(2);            // 返回 -1 (未找到) 

注意:

所有的值都在 [0, 1000000]的范围内。
操作的总数目在[1, 10000]范围内。
不要使用内建的哈希库。

class MyHashMap {
  /** Initialize your data structure here. */
        class Entry {
            int key;
            int value;
            Entry next;
            Entry(int key, int value) {
                this.key = key;
                this.value = value;
            }
        }
        int cap = 200000;
        Entry [] arr = new Entry[cap];
        public MyHashMap() {
        }

        /** value will always be non-negative. */
        public void put(int key, int value) {
            if(arr[key % cap] != null) {
                boolean isFound = false;
                Entry entry = arr[key % cap], last = null;
                while(entry != null) {
                    last = entry;
                    if(entry.key == key) {
                        entry.value = value;
                        isFound = true;
                        break;
                    }
                    entry = entry.next;
                }
                if(!isFound) {
                    last.next = new Entry(key, value);
                }
            } else {
                arr[key % cap] = new Entry(key, value);
            }
        }

        /** Returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key */
        public int get(int key) {
            Entry entry = arr[key % cap];
            while(entry != null) {
                if(entry.key == key) {
                    return entry.value;
                }
                entry = entry.next;
            }
            return -1;
        }

        /** Removes the mapping of the specified value key if this map contains a mapping for the key */
        public void remove(int key) {
            Entry entry = arr[key % cap], prev = null;
            while(entry != null) {
                if(entry.key == key) {
                    if (prev == null) {
                        arr[key % cap] = entry.next;
                    } else {
                        prev.next = entry.next;
                    }
                    break;
                }
                prev = entry;
                entry = entry.next;
            }
        }
}

/**
 * Your MyHashMap object will be instantiated and called as such:
 * MyHashMap obj = new MyHashMap();
 * obj.put(key,value);
 * int param_2 = obj.get(key);
 * obj.remove(key);
 */
posted @ 2020-04-07 17:23  南墙1  阅读(167)  评论(0编辑  收藏  举报