leetcode笔记之[706. 设计哈希映射]

51 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第23天,点击查看活动详情

题目描述

原题链接 :

706. 设计哈希映射

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

实现 MyHashMap 类:

  • MyHashMap() 用空映射初始化对象
  • void put(int key, int value) 向 HashMap 插入一个键值对 (key, value) 。如果 key 已经存在于映射中,则更新其对应的值 value 。
  • int get(int key) 返回特定的 key 所映射的 value ;如果映射中不包含 key 的映射,返回 -1 。
  • void remove(key) 如果映射中存在 key 的映射,则移除 key 和它所对应的 value 。  

示例:

输入:
["MyHashMap", "put", "put", "get", "get", "put", "get", "remove", "get"]
[[], [1, 1], [2, 2], [1], [3], [2, 1], [2], [2], [2]]
输出:
[null, null, null, 1, -1, null, 1, null, -1]

解释:
MyHashMap myHashMap = new MyHashMap();
myHashMap.put(1, 1); // myHashMap 现在为 [[1,1]]
myHashMap.put(2, 2); // myHashMap 现在为 [[1,1], [2,2]]
myHashMap.get(1);    // 返回 1 ,myHashMap 现在为 [[1,1], [2,2]]
myHashMap.get(3);    // 返回 -1(未找到),myHashMap 现在为 [[1,1], [2,2]]
myHashMap.put(2, 1); // myHashMap 现在为 [[1,1], [2,1]](更新已有的值)
myHashMap.get(2);    // 返回 1 ,myHashMap 现在为 [[1,1], [2,1]]
myHashMap.remove(2); // 删除键为 2 的数据,myHashMap 现在为 [[1,1]]
myHashMap.get(2);    // 返回 -1(未找到),myHashMap 现在为 [[1,1]]

提示:

  • 0 <= key, value <= 10^6
  • 最多调用 104 次 put、get 和 remove 方法

思路分析

开始想的是直接用两个数组来保存key和value的值。结果不行。 看了解题发现重要的是哈希, 用key%len 。 来确定 用vector<Node*> 来保存数据。 Node类似一个链表的结构。把映射值相同的key放到同一条链表中。 映射值用key%len来确定。 这里规定len = 1009;

所有的步骤都是先找到映射的链表头。 然后再链表中找key。

删除节点这里可以偷个懒、直接把k对应的v设为-1.

AC 代码

class Node {
public:
    int k;
    int v;
    Node* next;
    Node(int key, int value):k(key),v(value),next(nullptr) {}
};

int len = 1009;

class MyHashMap {
public:
    /** Initialize your data structure here. */
    MyHashMap() {
        arr = vector<Node*> (len, new Node(-1,-1));
    }
    
    /** value will always be non-negative. */
    void put(int key, int value) {
        int temp = key%len;
        Node* h = arr[temp];
        Node* p;
        while(h){
            if(h->k==key){
                h->v = value;
                return;
            }
            p = h;
            h = h->next;
        }
        Node* node = new Node(key, value);
        p->next = node;
    }
    
    /** Returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key */
    int get(int key) {
        int temp = key%len;
        Node* h = arr[temp];
        while(h){
            if(h->k==key) return h->v;
            h = h->next;
        }
        return -1;
        
    }
    
    /** Removes the mapping of the specified value key if this map contains a mapping for the key */
    void remove(int key) {
        int temp = key%len;
        Node* h = arr[temp];
        while(h){
            if(h->k==key){
                h->v = -1;
            }
            h = h->next;
        }
    }
private:
    vector<Node*> arr;
};

/**
 * 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);
 */