迭代器设计 #255
迭代器设计 #255
Conversation
internal/tree/red_black_tree.go
Outdated
| @@ -79,6 +104,15 @@ func newRBNode[K any, V any](key K, value V) *rbNode[K, V] { | |||
| } | |||
| } | |||
|
|
|||
| // 获取起始点的迭代器 | |||
| func (rb *RBTree[K, V]) Begin() iterator.Iterator[pair.Pair[K, V]] { | |||
There was a problem hiding this comment.
在类似树形结构里面，可以实现多种迭代器，分别按照不同的迭代顺序。比如说中序遍历，先序遍历等等……不同方法返回不同的迭代器。
|
我不太能接受这种设计。我认为 RBTree 不应该定义 DeleteIt 和 FindIt 这一类的方法。当我操作 Iterator 的时候，我希望我操作的就是 Iterator，而不是我还需要把这个东西传递回去 RBTree。 |
那我按照你的建议再改改 |
已经按照你的建议修改。 |
| @@ -95,6 +134,14 @@ func (rb *RBTree[K, V]) Delete(key K) (V, bool) { | |||
| return v, false | |||
| } | |||
|
|
|||
| // 查找结点 （但是返回iterator） | |||
There was a problem hiding this comment.
这个方法不需要。因为正常在使用 Iterator 的时候，查找是这么用的：
for itr.Next() {
val := itr.Get()
if val == xxx {
// 这就是判定
}
}There was a problem hiding this comment.
这样的话 iterator无法支持查找欸，比如我想使用iterator实现一个功能就是查找到比 key 大 和 比 key 小的 10 个值，本来可以做到O(logN)的现在只能做到O(N)
There was a problem hiding this comment.
Iterator 本身的语义就是迭代。Find 类的方法可以在 RBTree 上实现，但是没有必要在 Iterator 这里。
| } | ||
|
|
||
| func (iter *rbTreeIterator[K, V]) Next() bool { | ||
| fmt.Println(iter.currNode.key, iter.currNode.value) |
|
|
||
| func (iter *rbTreeIterator[K, V]) Next() bool { | ||
| fmt.Println(iter.currNode.key, iter.currNode.value) | ||
| iter.err = nil |
There was a problem hiding this comment.
这个 err 暂时不要置为 nil，如果要是已经不为 nil 了，你就永远返回 false。
There was a problem hiding this comment.
或者说，我们认为，如果 err 不为 nil，那么就说明这个 iterator 不可用了。
| @@ -53,6 +59,26 @@ type rbNode[K any, V any] struct { | |||
| left, right, parent *rbNode[K, V] | |||
| } | |||
|
|
|||
| func (node *rbNode[K, V]) getNext() *rbNode[K, V] { | |||
There was a problem hiding this comment.
不要定义在这里。你想，如果我们是广度优先遍历，那么 next 的概念是完全不一样的。你应该把这个挪过去 Iterator 的实现里面。
There was a problem hiding this comment.
确实是一个中序遍历，但是我有一个疑问就是对于一个红黑树来说，我们需要广度优先遍历吗，或者说我们需要除了中序遍历 (Begin 从小到大) 和反向中序遍历 (RBegin 从大到小)以外的遍历方式么
There was a problem hiding this comment.
你不需要纠结要不要，只是说我们需要留出来这个扩展点。所以你现在并不能直接说我这里不可能用别的遍历方式。而且你从设计理念上来说，next 就是归属你 iterator 的概念，一个红黑树是没有这个概念的。
| return vis > 0 | ||
| } | ||
|
|
||
| type rbTreeIterator[K any, V any] struct { |
There was a problem hiding this comment.
改个名字，名字里面包含它是如何遍历的，我的理解是你这个是中序遍历的？
There was a problem hiding this comment.
后续万一要实现广度优先、前序、后序，名字就能够区分出来。
There was a problem hiding this comment.
把这个 iterator 挪出去作为一个单独的文件，测试也要相应的挪出去。
| iter.nxtNode = iter.currNode.getNext() | ||
| return true | ||
| } | ||
| iter.err = ErrRBTreeIteratorNoNext |
| } | ||
|
|
||
| func (iter *rbTreeIterator[K, V]) Get() (kvPair pair.Pair[K, V]) { | ||
| iter.err = nil |
| return curr | ||
| } | ||
|
|
||
| func (rb *RBTree[K, V]) Begin() iterator.Iterator[pair.Pair[K, V]] { |
There was a problem hiding this comment.
同样将 Begin 改个名字，指明你的遍历顺序，在注释里面也解释一下。
| // 获取迭代器当前所指向的节点的信息 | ||
| Get() T | ||
|
|
||
| // 获取error |
There was a problem hiding this comment.
获取 error。当你在执行 Get 或者 Delete 之后，如果有必要，可以通过该方法来检测是否运作正常。
目前实现了Iterator接口以及接口在RBTree中所需要的新方法和使用方式，其他容器的接口实现方法也按照这样设计