Julia中遍历AbstractTrees时修改节点无法更新原结构的问询

当使用AbstractTrees遍历Julia表达式树时，直接修改循环中的node变量无法更新原树结构的核心原因是：

• 叶子节点（如Symbol、Number）是不可变值类型，遍历中拿到的是副本，修改局部变量不会影响原树；
• Expr是引用类型，修改其args数组会直接作用于原对象，因此能生效。

以下是两种可靠的修改方案：

方案1：递归构建新表达式树（纯函数式） #

适合不需要原地修改原表达式的场景，通过递归遍历生成全新的修改后表达式：

using AbstractTrees

mainexp = :(a + b * c)

function modify_expr(expr)
    # 处理叶子节点：替换:b为:d
    if expr isa Symbol
        return expr == :b ? :d : expr
    # 处理Expr节点：替换乘法*为÷
    elseif expr isa Expr
        new_args = [modify_expr(arg) for arg in expr.args]
        new_head = expr.head == :call && expr.args[1] == :* ? :÷ : expr.head
        return Expr(new_head, new_args...)
    # 保留其他类型（如数字）不变
    else
        return expr
    end
end

# 生成修改后的新表达式，原mainexp不受影响
modified_exp = modify_expr(mainexp)
# 输出结果：:(a + d ÷ c)

方案2：原地修改原表达式树（跟踪父节点与索引） #

如果需要直接修改原表达式结构，可在遍历过程中记录每个节点的父节点及其在父节点args中的索引，通过修改父节点的args数组实现叶子节点的替换：

using AbstractTrees

mainexp = :(a + b * c)

function modify_expr_inplace!(expr)
    # 用PostOrderDFS遍历，同时获取节点的状态信息（包含父节点与索引）
    for (node, state) in PostOrderDFS(expr)
        # 替换叶子节点:b为:d
        if node isa Symbol && node == :b
            parent_node = state.parent.node
            idx = state.index
            parent_node isa Expr && (parent_node.args[idx] = :d)
        # 替换乘法运算符*为÷
        elseif node isa Expr && node.head == :call && node.args[1] == :*
            node.args[1] = :÷
        end
    end
end

# 原地修改原表达式
modify_expr_inplace!(mainexp)
# 此时mainexp已变为：:(a + d ÷ c)

方案3：基于路径索引的修改（手动指定节点位置） #

如果已缓存了节点的路径索引（如从根节点到目标节点的args索引序列），可通过遍历路径直接定位并修改父节点的args：

mainexp = :(a + b * c)

function set_node!(root, path, new_val)
    current = root
    # 遍历到目标节点的父节点
    for idx in path[1:end-1]
        current = current.args[idx]
    end
    # 修改父节点args中对应位置的元素
    current.args[path[end]] = new_val
end

# 路径[2,1]表示：root.args[2]（即:(b*c)）的args[1]（即:b）
set_node!(mainexp, [2, 1], :d)
# 原mainexp变为：:(a + d * c)

注意事项 #

• 避免使用eval执行动态生成的路径表达式（如eval(:(mainexp.args[2].args[1] = :d))），这种方式存在作用域风险且代码可读性差；
• 对于复杂表达式树，优先使用递归构建新树或跟踪父节点的方式，逻辑更清晰且不易出错。

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