在 Swift 中实现简单树结构的递归生成器

Question

我在内存中有一个基于 XML 文档的简单树结构，我正在尝试编写一个递归生成器来支持 SequenceType，但我不知道如何实际执行此操作。

这是我的第一次尝试：

@objc public class XMLNode: NSObject, SequenceType {
    public weak var parentNode: XMLNode?
    public var nodeName: String
    public var attributes: [String: String]
    public var childNodes = [XMLNode]()

    public func generate() -> AnyGenerator<XMLNode> {
        var childGenerator = childNodes.generate()
        var returnedSelf = false

        return anyGenerator {
            let child = childGenerator.next()

            if child != nil {
                // I need to somehow recurse on child here

                return child
            } else if !returnedSelf {
                returnedSelf = true
                return self
            } else {
                return nil
            }
        }
    }
}

因为 childNodes 是一个数组，我调用它自己的 built-in generate() 函数在 child 节点上创建一个生成器并迭代它，然后最后返回 self。问题是它不会在每个 child 上递归，所以它只会深入一层。我不知道如何以这种方式组合两个生成器。

我很难思考如何做到这一点！制作递归生成器需要做什么？

Answer 1

~~我不知道生成器本身是否可以递归。~~ 请问M!

这是一个可能的先序遍历实现，对仍需枚举的子节点使用堆栈：

extension XMLNode : SequenceType {
    public func generate() -> AnyGenerator<XMLNode> {
        var stack : [XMLNode] = [self]
        return anyGenerator {
            if let next = stack.first {
                stack.removeAtIndex(0)
                stack.insertContentsOf(next.childNodes, at: 0)
                return next
            }
            return nil
        }
    }
}

对于层序遍历，替换

stack.insertContentsOf(next.childNodes, at: 0)

来自

stack.appendContentsOf(next.childNodes)

Answer 2

这是一个递归的 post 顺序生成器。不能说我会推荐实际使用它。 @MartinR 的回答似乎更实用一些

 public func generate() -> AnyGenerator<XMLNode> {
    var childGenerator:AnyGenerator<XMLNode>?
    var childArrayGenerator:IndexingGenerator<[XMLNode]>? = self.childNodes.generate()
    var returnedSelf = false

    return anyGenerator {
        if let next = childGenerator?.next() {
            return next
        }
        if let child = childArrayGenerator?.next() {
            childGenerator = child.generate()
            return childGenerator?.next()

        } else if !returnedSelf {

            returnedSelf = true
            return self

        } else {

            return nil
        }
    }
}

Answer 3

虽然 Martin 的答案肯定更简洁，但它的缺点是大量使用 array/insert 操作，并且在惰性序列操作中不是特别有用。这种替代方案应该适用于那些环境，我已经为 UIView 层次结构使用了类似的东西。

public typealias Generator = AnyGenerator<XMLNode>

public func generate() -> AnyGenerator<XMLNode> {
    var childGenerator = childNodes.generate()
    var subGenerator : AnyGenerator<XMLNode>?
    var returnedSelf = false

    return anyGenerator {
        if !returnedSelf {
            returnedSelf = true
            return self
        }

        if let subGenerator = subGenerator,
            let next = subGenerator.next() {
                return next
        }

        if let child = childGenerator.next() {
            subGenerator = child.generate()
            return subGenerator!.next()
        }

        return nil
    }
}

请注意，这是预购迭代，您可以为 post 订单移动 if !returnedSelf 块。

在 Swift 中实现简单树结构的递归生成器

Implementing recursive generator for simple tree structure in Swift

recursion

generator

swift

swift2