Ruby数组&运算的反向操作：保留重复元素的差集实现

嘿，这个场景我之前也碰到过！Ruby自带的Array#-方法在简单无重复的数组里用着很顺手，但它的逻辑是把第一个数组里所有在第二个数组中出现过的元素全删掉——这就是为什么[0,0,1] - [0]会返回[1]，它直接移除了所有的0，而不是只去掉和交集数量对应的那一个。

要实现“只移除与交集匹配次数相等的元素”（也就是让结果保留第一个数组里没被交集消耗掉的重复项），这里有几个简洁的原生解决方案：

方案1：用tally统计频次（Ruby 2.7+ 推荐） #

Ruby 2.7新增的tally方法可以快速统计数组中每个元素的出现次数，我们可以基于这个来计算剩余元素的数量，再展开成数组：

array1 = [0, 0, 1]
array2 = [0]

# 统计两个数组的元素频次
counts1 = array1.tally
counts2 = array2.tally

# 遍历统计结果，计算每个元素剩余的数量并展开
result = counts1.flat_map do |elem, total_count|
  # 减去交集里的出现次数，最少保留0个
  remaining = total_count - counts2.fetch(elem, 0)
  [elem] * remaining if remaining > 0
end.compact

p result # => [0, 1]

这个方法效率很高（时间复杂度O(n+m)），代码也清晰，适合现代Ruby版本。如果array2里有重复元素，比如array2 = [0,0]，它会自动移除array1里的两个0，返回[1]，完全符合预期。

方案2：手动遍历移除（兼容旧版本Ruby） #

如果你的Ruby版本低于2.7，没有tally方法，也可以手动遍历第二个数组，逐个从第一个数组的副本中移除匹配的元素（注意要操作副本，别修改原数组）：

array1 = [0, 0, 1]
array2 = [0]

# 复制原数组避免修改原始数据
temp_arr = array1.dup
array2.each do |elem|
  # 只移除第一个找到的匹配元素
  temp_arr.delete_at(temp_arr.index(elem)) if temp_arr.include?(elem)
end

p temp_arr # => [0, 1]

这个逻辑非常直观，而且不需要依赖新特性，适合所有Ruby版本。

方案3：第三方gem（可选） #

如果你不想自己写逻辑，也可以用difference这类第三方gem，它封装了保留重复项的差集方法，但毕竟是外部依赖，除非项目已经在用，否则原生实现会更轻量。

总结一下，最简洁高效的首选方案是用tally结合flat_map，代码短且逻辑清晰。如果是旧版本Ruby，手动遍历的方法也完全够用。

内容的提问来源于stack exchange，提问作者David B.