在Objective-C中从根到所有孩子遍历树

Question

我想做的一个简化版本是像这样转动一棵树：

变成这样的数组：["abc","abd","ae"]

基本上我想遍历树从它的根节点到每个孩子。

我尝试通过在递归块中放置一个 for-in 循环来做到这一点，但问题是每次递归块时 for 循环都会重新开始。当我尝试异步运行该块时，我不断收到 EXC_BAD_ACCESS

有什么建议吗？

Answer 1

假设树是这样表示的：

@interface TreeNode : NSObject
@property(weak,nonatomic) TreeNode *parent;
@property(strong,nonatomic) NSArray *children;
@end

任何节点的沿袭（这是您要查找的）是从根到节点的节点列表。这可以像这样递归定义：

- (NSArray *)lineage {
    if (!self.parent) {
        return @[self];
    } else {
        NSMutableArray *lineage = [[self.parent lineage] mutableCopy];
        [lineage addObject:self];
        return lineage;
    }
}

您正在寻找叶子的谱系，因此我们需要一种收集叶子的方法。如果我们可以遍历树，我们就可以做到这一点。这是一个很好的块应用程序，如下所示：

- (void)depthFirst:(void (^)(TreeNode *))block {
    for (TreeNode *node in self.children) {
        [node depthFirst:block];
    }
    return block(self);
}

这提供了一种自然的方式来收集树叶：

- (NSArray *)leaves {
    NSMutableArray *leaves = [@[] mutableCopy];
    [self depthFirst:^(TreeNode *node) {
        if (!node.children) [leaves addObject:node];
    }];
    return leaves;
}

把它放在一起，我们得到：

- (NSArray *)lineagesOfLeaves {
    NSMutableArray lineages = [@[] mutableCopy];
    for (TreeNode *leaf in [self leaves]) {
        [lineages addObject:[leaf lineage]];
    }
    return lineages;
}

这些方法适用于树中的任何节点。不过，对于您的问题，您需要将 lineagesOfLeaves 发送到树的根。