Skip to content

1.简单实现

js
/* 键值对 Number -> String */
class Pair {
  constructor(key, val) {
    this.key = key
    this.val = val
  }
}

/* 基于数组实现的哈希表 */
class ArrayHashMap {
  #buckets
  constructor() {
    // 初始化数组,包含 100 个桶
    this.#buckets = new Array(100).fill(null)
  }

  /* 哈希函数 */
  #hashFunc(key) {
    return key % 100
  }

  /* 查询操作 */
  get(key) {
    let index = this.#hashFunc(key)
    let pair = this.#buckets[index]
    if (pair === null) return null
    return pair.val
  }

  /* 添加操作 */
  set(key, val) {
    let index = this.#hashFunc(key)
    this.#buckets[index] = new Pair(key, val)
  }

  /* 删除操作 */
  delete(key) {
    let index = this.#hashFunc(key)
    // 置为 null ,代表删除
    this.#buckets[index] = null
  }

  /* 获取所有键值对 */
  entries() {
    let arr = []
    for (let i = 0; i < this.#buckets.length; i++) {
      if (this.#buckets[i]) {
        arr.push(this.#buckets[i])
      }
    }
    return arr
  }

  /* 获取所有键 */
  keys() {
    let arr = []
    for (let i = 0; i < this.#buckets.length; i++) {
      if (this.#buckets[i]) {
        arr.push(this.#buckets[i].key)
      }
    }
    return arr
  }

  /* 获取所有值 */
  values() {
    let arr = []
    for (let i = 0; i < this.#buckets.length; i++) {
      if (this.#buckets[i]) {
        arr.push(this.#buckets[i].val)
      }
    }
    return arr
  }

  /* 打印哈希表 */
  print() {
    let pairSet = this.entries()
    for (const pair of pairSet) {
      console.info(`${pair.key} -> ${pair.val}`)
    }
  }
}

2.哈希冲突

2-1.链式地址

js
/* 链式地址哈希表 */
class HashMapChaining {
  #size; // 键值对数量
  #capacity; // 哈希表容量
  #loadThres; // 触发扩容的负载因子阈值
  #extendRatio; // 扩容倍数
  #buckets; // 桶数组

  /* 构造方法 */
  constructor() {
    this.#size = 0;
    this.#capacity = 4;
    this.#loadThres = 2.0 / 3.0;
    this.#extendRatio = 2;
    this.#buckets = new Array(this.#capacity).fill(null).map((x) => []);
  }

  /* 哈希函数 */
  #hashFunc(key) {
    return key % this.#capacity;
  }

  /* 负载因子 */
  #loadFactor() {
    return this.#size / this.#capacity;
  }

  /* 查询操作 */
  get(key) {
    const index = this.#hashFunc(key);
    const bucket = this.#buckets[index];
    // 遍历桶,若找到 key ,则返回对应 val
    for (const pair of bucket) {
      if (pair.key === key) {
        return pair.val;
      }
    }
    // 若未找到 key ,则返回 null
    return null;
  }

  /* 添加操作 */
  put(key, val) {
    // 当负载因子超过阈值时,执行扩容
    if (this.#loadFactor() > this.#loadThres) {
      this.#extend();
    }
    const index = this.#hashFunc(key);
    const bucket = this.#buckets[index];
    // 遍历桶,若遇到指定 key ,则更新对应 val 并返回
    for (const pair of bucket) {
      if (pair.key === key) {
        pair.val = val;
        return;
      }
    }
    // 若无该 key ,则将键值对添加至尾部
    const pair = new Pair(key, val);
    bucket.push(pair);
    this.#size++;
  }

  /* 删除操作 */
  remove(key) {
    const index = this.#hashFunc(key);
    let bucket = this.#buckets[index];
    // 遍历桶,从中删除键值对
    for (let i = 0; i < bucket.length; i++) {
      if (bucket[i].key === key) {
        bucket.splice(i, 1);
        this.#size--;
        break;
      }
    }
  }

  /* 扩容哈希表 */
  #extend() {
    // 暂存原哈希表
    const bucketsTmp = this.#buckets;
    // 初始化扩容后的新哈希表
    this.#capacity *= this.#extendRatio;
    this.#buckets = new Array(this.#capacity).fill(null).map((x) => []);
    this.#size = 0;
    // 将键值对从原哈希表搬运至新哈希表
    for (const bucket of bucketsTmp) {
      for (const pair of bucket) {
        this.put(pair.key, pair.val);
      }
    }
  }

  /* 打印哈希表 */
  print() {
    for (const bucket of this.#buckets) {
      let res = [];
      for (const pair of bucket) {
        res.push(pair.key + " -> " + pair.val);
      }
      console.log(res);
    }
  }
}

2-2.线性探测

js
/* 开放寻址哈希表 */
class HashMapOpenAddressing {
  #size; // 键值对数量
  #capacity; // 哈希表容量
  #loadThres; // 触发扩容的负载因子阈值
  #extendRatio; // 扩容倍数
  #buckets; // 桶数组
  #TOMBSTONE; // 删除标记

  /* 构造方法 */
  constructor() {
    this.#size = 0; // 键值对数量
    this.#capacity = 4; // 哈希表容量
    this.#loadThres = 2.0 / 3.0; // 触发扩容的负载因子阈值
    this.#extendRatio = 2; // 扩容倍数
    this.#buckets = Array(this.#capacity).fill(null); // 桶数组
    this.#TOMBSTONE = new Pair(-1, "-1"); // 删除标记
  }

  /* 哈希函数 */
  #hashFunc(key) {
    return key % this.#capacity;
  }

  /* 负载因子 */
  #loadFactor() {
    return this.#size / this.#capacity;
  }

  /* 搜索 key 对应的桶索引 */
  #findBucket(key) {
    let index = this.#hashFunc(key);
    let firstTombstone = -1;
    // 线性探测,当遇到空桶时跳出
    while (this.#buckets[index] !== null) {
      // 若遇到 key ,返回对应的桶索引
      if (this.#buckets[index].key === key) {
        // 若之前遇到了删除标记,则将键值对移动至该索引处
        if (firstTombstone !== -1) {
          this.#buckets[firstTombstone] = this.#buckets[index];
          this.#buckets[index] = this.#TOMBSTONE;
          return firstTombstone; // 返回移动后的桶索引
        }
        return index; // 返回桶索引
      }
      // 记录遇到的首个删除标记
      if (firstTombstone === -1 && this.#buckets[index] === this.#TOMBSTONE) {
        firstTombstone = index;
      }
      // 计算桶索引,越过尾部则返回头部
      index = (index + 1) % this.#capacity;
    }
    // 若 key 不存在,则返回添加点的索引
    return firstTombstone === -1 ? index : firstTombstone;
  }

  /* 查询操作 */
  get(key) {
    // 搜索 key 对应的桶索引
    const index = this.#findBucket(key);
    // 若找到键值对,则返回对应 val
    if (
      this.#buckets[index] !== null &&
      this.#buckets[index] !== this.#TOMBSTONE
    ) {
      return this.#buckets[index].val;
    }
    // 若键值对不存在,则返回 null
    return null;
  }

  /* 添加操作 */
  put(key, val) {
    // 当负载因子超过阈值时,执行扩容
    if (this.#loadFactor() > this.#loadThres) {
      this.#extend();
    }
    // 搜索 key 对应的桶索引
    const index = this.#findBucket(key);
    // 若找到键值对,则覆盖 val 并返回
    if (
      this.#buckets[index] !== null &&
      this.#buckets[index] !== this.#TOMBSTONE
    ) {
      this.#buckets[index].val = val;
      return;
    }
    // 若键值对不存在,则添加该键值对
    this.#buckets[index] = new Pair(key, val);
    this.#size++;
  }

  /* 删除操作 */
  remove(key) {
    // 搜索 key 对应的桶索引
    const index = this.#findBucket(key);
    // 若找到键值对,则用删除标记覆盖它
    if (
      this.#buckets[index] !== null &&
      this.#buckets[index] !== this.#TOMBSTONE
    ) {
      this.#buckets[index] = this.#TOMBSTONE;
      this.#size--;
    }
  }

  /* 扩容哈希表 */
  #extend() {
    // 暂存原哈希表
    const bucketsTmp = this.#buckets;
    // 初始化扩容后的新哈希表
    this.#capacity *= this.#extendRatio;
    this.#buckets = Array(this.#capacity).fill(null);
    this.#size = 0;
    // 将键值对从原哈希表搬运至新哈希表
    for (const pair of bucketsTmp) {
      if (pair !== null && pair !== this.#TOMBSTONE) {
        this.put(pair.key, pair.val);
      }
    }
  }

  /* 打印哈希表 */
  print() {
    for (const pair of this.#buckets) {
      if (pair === null) {
        console.log("null");
      } else if (pair === this.#TOMBSTONE) {
        console.log("TOMBSTONE");
      } else {
        console.log(pair.key + " -> " + pair.val);
      }
    }
  }
}