Swift国际象棋游戏树实现中的Weak引用内存管理问题

核心问题分析 #

你的场景里，节点间的引用关系是：父节点持有子节点（next和variations）的强引用，子节点持有父节点的引用。如果子节点对父节点用强引用，确实会形成循环引用（父→子→父），导致内存无法释放；但直接给parent加weak测试失效，是因为如果没有其他强引用持有父节点，子节点的weak parent会提前变成nil。

修正后的实现方案 #

正确的做法是：父节点强持有所有子节点，子节点对父节点使用weak引用，同时确保在遍历或GUI展示时，始终有一个强引用链持有整个树的根节点（或当前正在操作的节点分支）。这样既避免循环引用，又不会出现节点意外释放的问题。

修正后的ChessNode代码 #

class ChessNode {
    // 局面数据，简化表示
    let position: String
    // 父节点：弱引用，打破循环引用
    weak var parent: ChessNode?
    // 主变着子节点：强引用，父节点主导子节点生命周期
    var next: ChessNode?
    // 变着子节点列表：强引用
    var variations: [ChessNode] = []
    
    init(position: String) {
        self.position = position
    }
    
    // 添加主变着节点
    func addNext(position: String) -> ChessNode {
        let node = ChessNode(position: position)
        node.parent = self
        self.next = node
        return node
    }
    
    // 添加变着节点
    func addVariation(position: String) -> ChessNode {
        let node = ChessNode(position: position)
        node.parent = self
        self.variations.append(node)
        return node
    }
    
    // 节点跳转方法：回到父节点
    func goToParent() -> ChessNode? {
        return parent
    }
    
    // 跳转到主变着下一个节点
    func goToNext() -> ChessNode? {
        return next
    }
    
    // 跳转到指定变着节点
    func goToVariation(at index: Int) -> ChessNode? {
        guard index >= 0, index < variations.count else { return nil }
        return variations[index]
    }
}

测试用例验证 #

测试时需确保根节点始终被强引用持有，避免整个树被提前回收：

// 测试代码
func testChessTree() {
    // 根节点强引用，作为整个树的持有源
    let root = ChessNode(position: "初始局面")
    
    // 构建主变着链：初始→e4→e5→Nf3
    let e4 = root.addNext(position: "e4")
    let e5 = e4.addNext(position: "e5")
    let nf3 = e5.addNext(position: "Nf3")
    
    // 给e4添加变着：e4→c5（西西里防御）
    let c5 = e4.addVariation(position: "c5")
    
    // 测试跳转逻辑
    assert(nf3.goToParent() === e5)
    assert(e4.goToNext() === e5)
    assert(e4.goToVariation(at: 0) === c5)
    
    // 测试内存释放
    weak var weakE4 = e4
    weak var weakC5 = c5
    root.next = nil
    e4.variations.removeAll()
    // 根节点相关引用清除后，子节点应被释放
    assert(weakE4 == nil)
    assert(weakC5 == nil)
}

testChessTree()

关键注意事项 #

• 根节点的强引用：在GUI或业务逻辑中，必须有一个强引用变量持有当前树的根节点（或当前浏览分支的顶端节点），否则整个树会被系统回收。
• 变着节点的管理：variations用数组强持有变着节点，父节点释放时，这些变着节点会自动被释放，因为没有其他强引用。
• 循环引用的避免：子节点的parent是weak，打破了父→子→父的强引用循环，内存可以正常释放。

内容的提问来源于stack exchange，提问作者dehlen