1. 程式人生 > >LeetCode 706.設計雜湊對映(Design HashMap)

LeetCode 706.設計雜湊對映(Design HashMap)

不使用任何內建的雜湊表庫設計一個雜湊對映 

具體地說,你的設計應該包含以下的功能

  • put(key, value):向雜湊對映中插入(鍵,值)的數值對。如果鍵對應的值已經存在,更新這個值。
  • get(key):返回給定的鍵所對應的值,如果對映中不包含這個鍵,返回-1。
  • remove(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 (未找到) 

注意:

  • 所有的值都在 [1, 1000000]的範圍內。
  • 操作的總數目在[1, 10000]範圍內。
  • 不要使用內建的雜湊庫。

 此題和leetcode705相比,加了一個key,不過這題我用的是鏈地址法。

706C程式碼:

#define SIZE 50000
struct node {
    int key;
    int val;
    struct node* next;
};
typedef struct {
    struct node* head[SIZE];
} MyHashMap;

/** Initialize your data structure here. */
MyHashMap* myHashMapCreate() {
    MyHashMap* obj = (MyHashMap*)malloc(sizeof(MyHashMap));
    for (int i = 0;i < SIZE;i++)
        obj->head[i] = NULL;
    return obj;
}
void creat(MyHashMap* obj,int key,int value)
{
    struct node* p = (struct node*)malloc(sizeof(struct node));
    p->key = key;
    p->val = value;
    p->next = obj->head[key%SIZE];
    obj->head[key%SIZE] = p;
}
/** value will always be positive. */
void myHashMapPut(MyHashMap* obj, int key, int value) {
    int t = key % SIZE;
    if (obj->head[t] == NULL)
    {
        creat(obj,key,value);
    }
    else
    {
        struct node* p = obj->head[t];
        while (p && p->key != key)  p = p->next;
        if (p == NULL)
        {
            creat(obj,key,value);
        }
        else
        {
            p->val = value;
        }
    }
}

/** Returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key */
int myHashMapGet(MyHashMap* obj, int key) {
    int t = key % SIZE;
    if (obj->head[t] != NULL)
    {
        struct node* p = obj->head[t];
        while (p && p->key != key)  p = p->next;
        if (p->key == key)
            return p->val;
    }
    return -1;
}

/** Removes the mapping of the specified value key if this map contains a mapping for the key */
void myHashMapRemove(MyHashMap* obj, int key) {
    int t = key % SIZE;
    if (obj->head[t] != NULL)
    {
        struct node* p = obj->head[t];
        struct node* fp = p;
        while (p && p->key != key)
        {
            fp = p;
            p = p->next;
        }
        if (p != NULL && fp != p)
            fp->next = p->next;
        if (fp == p)
            obj->head[t] = p->next;
    }
}

void myHashMapFree(MyHashMap* obj) {
    for (int i = 0;i < SIZE;i++)
    {
        struct node* p = obj->head[i];
        while (p)
        {
            struct node* q = p->next;
            free(p);
            p = q;
        }
    }
    free(obj);
}

/**
 * Your MyHashMap struct will be instantiated and called as such:
 * struct MyHashMap* obj = myHashMapCreate();
 * myHashMapPut(obj, key, value);
 * int param_2 = myHashMapGet(obj, key);
 * myHashMapRemove(obj, key);
 * myHashMapFree(obj);
 */

謝謝。