-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMyHashMap.cs
85 lines (70 loc) · 2.33 KB
/
MyHashMap.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
//https://leetcode.com/problems/design-hashmap
namespace LeetCode.Problems;
public sealed class MyHashMap : ProblemBase
{
[Theory]
[ClassData(typeof(MyHashMap))]
public override void Test(object[] data) => base.Test(data);
protected override void AddTestCases()
=> Instructions<HashMap, int[]>(config =>
config
.MapConstructor("MyHashMap")
.MapInstruction("put", (it, data) => it.Put(data[0], data[1]))
.MapInstruction("get", (it, data) => it.Get(data[0]))
.MapInstruction("remove", (it, data) => it.Remove(data[0]))
)
.Add(it => it.Data<int>("""[[], [1, 1], [2, 2], [1], [3], [2, 1], [2], [2], [2]]""")
.Instructions("""["MyHashMap", "put", "put", "get", "get", "put", "get", "remove", "get"]""")
.Output("[null, null, null, 1, -1, null, 1, null, -1]"));
public class HashMap
{
class Data
{
public int Key;
public int Value;
}
private readonly LinkedList<Data>?[] _data = new LinkedList<Data>?[1000];
public void Put(int key, int value)
{
var index = key % _data.Length;
_data[index] ??= new LinkedList<Data>();
var node = _data[index]!.First;
while (node != null)
{
if (node.Value.Key == key)
{
node.Value.Value = value;
return;
}
node = node.Next;
}
_data[index]!.AddFirst(new Data {Key = key, Value = value});
}
public int Get(int key)
{
var node = GetNode(key);
return node != null ? node.Value.Value : -1;
}
public void Remove(int key)
{
var node = GetNode(key);
if (node != null)
{
_data[key % _data.Length]!.Remove(node);
}
}
private LinkedListNode<Data>? GetNode(int key)
{
var node = _data[key % _data.Length]?.First;
while (node != null)
{
if (node.Value.Key == key)
{
return node;
}
node = node.Next;
}
return null;
}
}
}