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
|
// RUN: not %target-swift-frontend %s -emit-ir
public struct Trie<Key: Hashable, Value> {
internal var root: TrieNode<Key, Value>
public init() {
self.root = TrieNode<Key, Value>()
}
public mutating func removeValue<C: Collection>(forCollection collection: C) -> Value? where C.Iterator.Element == Key {
return self.root.removeValue(forIterator: collection.makeIterator())
}
public mutating func updateValue<C: Collection>(_ value: Value, forCollection collection: C) -> Value? where C.Iterator.Element == Key {
return self.root.updateValue(value, forIterator: collection.makeIterator())
}
public func value<C: Collection>(forCollection collection: C) -> Value? where C.Iterator.Element == Key {
return self.root.value(forIterator: collection.makeIterator())
}
}
internal struct TrieNode<Key: Hashable, Value> {
internal let children: Dictionary<Key, TrieNode<Key, Value>>
internal let value: Value?
internal init(value: Value?=nil) {
self.children = Dictionary<Key, TrieNode<Key, Value>>()
self.value = value
}
internal mutating func removeValue<I: IteratorProtocol>(forIterator iterator: inout I) -> Value? where I.Element == Key {
return nil
}
internal mutating func updateValue<I: IteratorProtocol>(_ value: Value, forIterator iterator: inout I) -> Value? where I.Element == Key {
return nil
}
internal func value<I: IteratorProtocol>(forIterator iterator: inout I) -> Value? where I.Element == Key {
guard let key = iterator.next() else {
return value
}
return self.children[key]?.value(forIterator: iterator)
}
}
|
