天天看點

LeetCode 706. 設計哈希映射(Java)

706. 設計哈希映射

不使用任何内建的哈希表庫設計一個哈希映射

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

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 (未找到)

注意:

所有的值都在 [0, 1000000]的範圍内。

操作的總數目在[1, 10000]範圍内。

不要使用内建的哈希庫。

來源:力扣(LeetCode)

連結:https://leetcode-cn.com/problems/design-hashmap

著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。

class Pair<U, V> {
  public U first;
  public V second;
  public Pair(U first, V second) {
    this.first = first;
    this.second = second;
  }
}
class Bucket {
  private List<Pair<Integer, Integer>> bucket;
  public Bucket() {
    this.bucket = new LinkedList<Pair<Integer, Integer>>();
  }
  public Integer get(Integer key) {
    for (Pair<Integer, Integer> pair : this.bucket) {
      if (pair.first.equals(key))
        return pair.second;
    }
    return -1;
  }
  public void update(Integer key, Integer value) {
    boolean found = false;
    for (Pair<Integer, Integer> pair : this.bucket) {
      if (pair.first.equals(key)) {
        pair.second = value;
        found = true;
      }
    }
    if (!found)
      this.bucket.add(new Pair<Integer, Integer>(key, value));
  }
  public void remove(Integer key) {
    for (Pair<Integer, Integer> pair : this.bucket) {
      if (pair.first.equals(key)) {
        this.bucket.remove(pair);
        break;
      }
    }
  }
}
class MyHashMap {
  private int key_space;
  private List<Bucket> hash_table;
  /** Initialize your data structure here. */
  public MyHashMap() {
    this.key_space = 2069;
    this.hash_table = new ArrayList<Bucket>();
    for (int i = 0; i < this.key_space; ++i) {
      this.hash_table.add(new Bucket());
    }
  }
  /** value will always be non-negative. */
  public void put(int key, int value) {
    int hash_key = key % this.key_space;
    this.hash_table.get(hash_key).update(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) {
    int hash_key = key % this.key_space;
    return this.hash_table.get(hash_key).get(key);
  }
  /** Removes the mapping of the specified value key if this map contains a mapping for the key */
  public void remove(int key) {
    int hash_key = key % this.key_space;
    this.hash_table.get(hash_key).remove(key);
  }
}
/**
 * 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);
 */