如何使用 Golang 函数自定义数据结构的迭代方式
在 Go 语言中,我们可以通过实现 Iterator 接口来自定义数据结构的迭代方式。以下是具体步骤:
1. 定义 Iterator 接口
Iterator 接口定义了用于遍历集合所需的方法。它只包含一个方法:
立即学习“go语言免费学习笔记(深入)”;
type Iterator interface {
Next() (value interface{}, ok bool)
}
- Next 方法返回集合中的下一个元素及其是否存在。
2. 实现 Iterator 接口
为了自定义数据结构的迭代方式,我们需要实现 Iterator 接口。以下是一个示例:
type MyIterator struct {
data []int
index int
}
func (i *MyIterator) Next() (value interface{}, ok bool) {
if i.index >= len(i.data) {
return nil, false
}
value = i.data[i.index]
i.index++
return value, true
}
3. 使用自定义迭代器
实现 Iterator 接口后,我们就可以像使用标准库迭代器一样使用自定义迭代器。例如,我们可以使用 range 关键字进行迭代:
for _, value := range MyIterator{[]int{1, 2, 3}} {
fmt.Println(value)
}
实战案例
以下是一个使用自定义迭代器遍历二叉树的实战案例:
type Node struct {
Value int
Left *Node
Right *Node
}
func (n *Node) Iterator() Iterator {
return &nodeIterator{n}
}
type nodeIterator struct {
node *Node
stack []*Node
}
func (i *nodeIterator) Next() (value interface{}, ok bool) {
for len(i.stack) > 0 || i.node != nil {
if i.node != nil {
i.stack = append(i.stack, i.node)
i.node = i.node.Left
} else {
i.node = i.stack[len(i.stack)-1]
value = i.node.Value
i.stack = i.stack[:len(i.stack)-1]
i.node = i.node.Right
return value, true
}
}
return nil, false
}
func main() {
root := &Node{Value: 1}
root.Left = &Node{Value: 2}
root.Right = &Node{Value: 3}
root.Left.Left = &Node{Value: 4}
root.Left.Right = &Node{Value: 5}
root.Right.Left = &Node{Value: 6}
root.Right.Right = &Node{Value: 7}
for _, value := range root.Iterator() {
fmt.Println(value)
}
}
这段代码使用迭代器实现了对二叉树的中序遍历。