本文共 2529 字，大约阅读时间需要 8 分钟。

Objective-C实现合并两棵二叉树算法示例

TreeNode类定义

#import 
   
    
@interface TreeNode : NSObject
@property NSInteger val;
@property TreeNode *left;
@property TreeNode *right;
@end
   

算法概述

在Objective-C中实现二叉树的合并算法，可以通过递归或迭代的方式完成。这两种方法各有优缺点，通常根据具体需求选择使用。以下将详细介绍实现过程。

递归实现方法

递归是一种简洁且易于理解的方法，适合处理二叉树问题。递归的思路是将问题分解为更小的子问题，最终达到叶节点。

迭代实现方法

迭代方法通过使用栈来模拟递归过程，避免了递归深度过大的问题。这种方法在处理较大树时效率更高。

核心代码示例

#import 
   
    
@interface TreeNode : NSObject
@property NSInteger val;
@property TreeNode *left;
@property TreeNode *right;
@end
@implementation TreeNode
+ (TreeNode *)mergeTwoTrees:(TreeNode *)tree1 :(TreeNode *)tree2 {
    if (!tree1 && !tree2) return nil;
    
    TreeNode *left = nil;
    TreeNode *right = nil;
    
    // 处理左子树
    if (tree1) left = [self mergeTwoTrees tree1.left tree2.left];
    if (tree2) right = [self mergeTwoTrees tree2.left tree2.right];
    
    TreeNode *current = [[TreeNode alloc] init];
    current.val = max(tree1.val, tree2.val);
    current.left = left;
    current.right = right;
    
    return current;
}
+ (TreeNode *)mergeTwoTrees:(TreeNode *)tree1 :(TreeNode *)tree2 {
    TreeNode *result = nil;
    
    // 迭代方法
    Stack *stack = [Stack new];
    [stack push: tree1];
    [stack push: tree2];
    [stack push: [TreeNode class]];
    
    while (![stack isEmpty]) {
        TreeNode *node = [stack pop];
        if ([node isKindOfClass: [TreeNode class]]) {
            // 创建新节点
            TreeNode *current = [[TreeNode alloc] init];
            current.val = [node isKindOfClass: [TreeNode class]] ? [node.val : 0] : [tree1.val : tree2.val];
            // 需要根据左、右情况继续处理
            if (!tree1 && !tree2) {
                result = current;
            } else if (node.left) {
                // 处理左子树
                [stack push: node.left];
            } else {
                // 处理右子树
                [stack push: node.right];
            }
        } else if ([node isKindOfClass: [Stack class]]) {
            // 栈标记处理
            if (stack.count >= 3) {
                // 处理节点
            }
        }
    }
    
    return result;
}
   

注意事项

通过以上方法，可以在Objective-C中实现两棵二叉树的合并。选择递归或迭代方法取决于具体需求和性能考虑。