forked from neetcode-gh/leetcode
- Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path0208-implement-trie-prefix-tree.cs
57 lines (50 loc) · 1.34 KB
/
0208-implement-trie-prefix-tree.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
publicclassTrieNode{
publicTrieNode(){
childrenMap=newDictionary<char,TrieNode>();
}
publicDictionary<char,TrieNode>childrenMap{get;set;}
publicboolisWord{get;set;}
}
publicclassTrie{
privateTrieNoderoot;
publicTrie(){
root=newTrieNode();
}
publicvoidInsert(stringword){
varcur=root;
foreach(varcinword){
if(!cur.childrenMap.ContainsKey(c)){
cur.childrenMap[c]=newTrieNode();
}
cur=cur.childrenMap[c];
}
cur.isWord=true;
}
publicboolSearch(stringword){
varnode=traverse(word);
returnnode!=null&&node.isWord;
}
publicboolStartsWith(stringprefix){
varnode=traverse(prefix);
returnnode!=null;
}
privateTrieNodetraverse(stringpath){
varcur=root;
foreach(varcinpath){
if(cur.childrenMap.ContainsKey(c)){
cur=cur.childrenMap[c];
}
else{
returnnull;
}
}
returncur;
}
}
/**
* Your Trie object will be instantiated and called as such:
* Trie obj = new Trie();
* obj.Insert(word);
* bool param_2 = obj.Search(word);
* bool param_3 = obj.StartsWith(prefix);
*/