如何使用 Golang 函数自定义数据结构的迭代方式

在 Go 语言中，我们可以通过实现 Iterator 接口来自定义数据结构的迭代方式。以下是具体步骤：

1. 定义 Iterator 接口

Iterator 接口定义了用于遍历集合所需的方法。它只包含一个方法：

立即学习“go语言免费学习笔记（深入）”；

type Iterator interface {
    Next() (value interface{}, ok bool)
}

• Next 方法返回集合中的下一个元素及其是否存在。

2. 实现 Iterator 接口

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

为了自定义数据结构的迭代方式，我们需要实现 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)
    }
}

这段代码使用迭代器实现了对二叉树的中序遍历。