目录
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 <= 106
- 最多调用
104
次put
、get
和remove
方法
实现代码:
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
class MyHashMap {
static class Pair{
int key, val;
Pair(int key, int val) {
this.key = key;
this.val = val;
}
}
private static int cap;
private LinkedList[] arr;
public MyHashMap() {
cap = 769;
arr = new LinkedList[cap];
for (int i = 0; i < cap; i++) {
arr[i] = new LinkedList();
}
}
public void put(int key, int value) {
LinkedList h = arr[hash(key)];
Iterator<Pair> iterator = h.iterator();
while (iterator.hasNext()) {
Pair pair = iterator.next();
if (key == pair.key) {
pair.val = value;
return;
}
}
h.offerLast(new Pair(key, value));
}
public int get(int key) {
LinkedList h = arr[hash(key)];
Iterator<Pair> iterator = h.iterator();
while (iterator.hasNext()) {
Pair pair = iterator.next();
if (key == pair.key ) {
return pair.val;
}
}
return -1;
}
public void remove(int key) {
LinkedList h = arr[hash(key)];
Iterator<Pair> iterator = h.iterator();
while (iterator.hasNext()) {
Pair pair = iterator.next();
if (key == pair.key ) {
h.remove(pair);
return;
}
}
}
private static int hash(int key) {
return key % cap;
}
}