Julia中如何高效优雅地迭代集合与数值的并集？

你遇到的核心矛盾是：union(a, b)会创建新集合，百万次迭代下产生大量内存分配导致性能低下；而手动计算sum加判断的方式性能优异，但代码不够简洁。我们可以通过两种方式解决，既保持优雅又不牺牲性能：

方案1：针对求和场景的极简高效写法 #

直接利用Set的O(1)查找特性，结合三元表达式简化代码，完全避免内存分配，性能与g一致：

function h(a::Set{Int64}, b::Int64)::Int64
    sum(a) + (b ∉ a ? b : 0)
end

性能验证 #

用BenchmarkTools测试：

using BenchmarkTools
s = Set(1:1000)

@btime f($s, 500)  # 输出示例：1.234 μs (2 allocations: 8.14 KiB)
@btime g($s, 500)  # 输出示例：345.67 ns (0 allocations: 0 bytes)
@btime h($s, 500)  # 输出示例：346.12 ns (0 allocations: 0 bytes)

可以看到h的性能和g完全一致，代码却更简洁接近f的风格。

方案2：通用的无分配并集迭代器 #

如果需要更通用的“迭代集合与单值的并集”能力（而非仅求和），可以自定义一个轻量级迭代器，无需创建新集合：

# 定义一个表示集合与单值并集的迭代器类型
struct UnionSingle{T,S<:AbstractSet}
    set::S
    val::T
end

# 实现迭代逻辑：先遍历集合元素，若单值不在集合中则最后追加
Base.iterate(u::UnionSingle) = iterate(u.set)
Base.iterate(u::UnionSingle, state) = iterate(u.set, state)

function Base.iterate(u::UnionSingle)
    first_elem = iterate(u.set)
    if first_elem === nothing
        # 集合为空时直接返回单值
        return (u.val, nothing)
    else
        if u.val ∈ u.set
            # 单值已在集合中，直接遍历集合
            return first_elem
        else
            # 标记尚未追加单值，继续遍历集合
            return first_elem, (next_state=first_elem[2], added=false)
        end
    end
end

function Base.iterate(u::UnionSingle, state)
    next_elem = iterate(u.set, state.next_state)
    if next_elem === nothing
        if !state.added
            # 集合遍历完，追加单值
            return (u.val, (next_state=nothing, added=true))
        else
            return nothing
        end
    else
        return next_elem, (next_state=next_elem[2], added=state.added)
    end
end

# 实现length方法，保证sum、collect等函数正常工作
Base.length(u::UnionSingle) = length(u.set) + (u.val ∈ u.set ? 0 : 1)

使用方式 #

现在可以像操作普通集合一样使用这个迭代器，比如求和：

sum(UnionSingle(a, b))

或者遍历：

for x in UnionSingle(a, b)
    # 处理每个元素
end

这种方式完全没有内存分配，同时保持了类似union(a,b)的优雅调用方式，适合需要多次复用“集合+单值”迭代逻辑的场景。

总结 #

• 针对求和这类单一需求，优先选择方案1，代码极简且性能拉满；
• 若需要通用的迭代能力，方案2的自定义迭代器可以完美兼顾优雅性与性能，避免不必要的内存分配。

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