如何防止重复引用数组中的类对象？

Question

我正在寻找一种方法来防止数组重复引用其中的同一对象。我的意思不是重复值 - 这很好。

例如，我知道在 Apple 的 SpriteKit 框架中，一个 SKNode 对象将其子对象存储在一个数组 (var children: [SKNode]) 中，两次向父对象添加一个节点会导致程序崩溃。

let parent = SKNode()
let child1 = SKNode()
let child2 = SKNode()
parent.addChild(child1)
parent.addChild(child2)    // Allowed, although child2 == child1.
parent.addChild(child1)    // Causes a crash.

这正是我想要模仿的行为。我将如何管理这个？是否有可能无需比较每个引用而产生 O(n) 复杂度？

Answer 1

我不确定您究竟为什么要这样做，但您可以这样做：

...

var childReferences = Set<ObjectIdentifier>

private func validateUniqueness(_ node: SKNode) {
    guard childReferences.insert(ObjectIdentifier(node)).inserted else {
        fatalError("An attempt was made to add a duplicate child")
    }
}

override func addChild(_ node: SKNode) {
    validateUniqueness(node)
    super.addChild(node)
}

override func insertChild(_ node: SKNode at index: Int) {
    validateUniqueness(node)
    super.insertChild(node, at: index)
}

override func removeChildren(in nodes: [SKNode]) {
    childReferences.subtract(nodes)
    super.removeChildren(in nodes: [SKNode])
}

override func removeAllChildren() {
    childReferences.removeAll(keepingCapacity: false)
    super.removeAllChildren()
}