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是一个数组，我调用自己的内置generate()函数在子节点上创建生成器并迭代它，然后返回self在末尾。问题是它并没有在每个孩子身上递归，所以它只会达到一个深度。我无法弄清楚如何以这种方式组合两台发电机。

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

Answer 1

~~我不知道发电机本身是否可以递归。 M proved me wrong！~~

这是一个可能的预执行遍历实现，使用堆栈为子节点仍然需要枚举：

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

这是一个递归的订单后生成器。不能说我建议实际使用它。 @ 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的答案肯定更简洁，但它有许多使用大量数组/插入操作的缺点，并且在延迟序列操作中并不是特别有用。这个替代方案应该适用于那些环境，我使用了类似于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
    }
}

请注意，这是预购迭代，您可以移动if !returnedSelf块以进行下单。

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

3 个答案: