Ruby判断数组中是否存在相同的:nb键值对

问题分析 #

你需要检查第一个数组中哈希元素的"nb"（注意示例中是字符串键，若实际为符号:nb需对应调整）值，是否在第二个数组的哈希元素中存在。由于数组包含数千个元素，必须优先考虑性能，避免O(n²)时间复杂度的实现。

原代码的问题 #

你当前的代码存在以下问题：

• 语法错误：arr1.map(&:nb) 无效，哈希对象没有nb实例方法，需通过键取值（如h["nb"]）。
• 逻辑混乱：代码块结构不完整，nb && val的逻辑不符合需求。
• 性能低下：每次调用arr2.map(...).include?都会重新遍历arr2生成临时数组，数千元素下会导致严重的性能损耗。

优化实现方案 #

利用Ruby的Set类（需引入set库），将arr2的"nb"值存入集合，集合的include?操作是O(1)时间复杂度，整体实现为O(m+n)的线性时间，适合大数据量场景。

步骤1：生成arr2的nb值集合

require 'set'

# 提取arr2中所有哈希的"nb"值，转为Set
arr2_nb_set = arr2.map { |hash| hash["nb"] }.to_set

步骤2：按需执行检查

根据你的具体需求，选择对应的代码：

• 找出arr1中nb值不在arr2里的元素：

  missing_items = arr1.reject { |hash| arr2_nb_set.include?(hash["nb"]) }
  # 示例输出: [{"nb"=>"5qwercv546", "active"=>true, "brand"=>"gem"}]
  

• 判断arr1中是否存在nb值不在arr2里的元素：

  has_missing = arr1.any? { |hash| !arr2_nb_set.include?(hash["nb"]) }
  # 示例输出: true
  

• 找出arr1中nb值在arr2里存在的元素：

  existing_items = arr1.select { |hash| arr2_nb_set.include?(hash["nb"]) }
  # 示例输出: arr1的前两个元素
  

性能说明 #

使用Set替代数组的include?，避免了每次检查都遍历整个arr2的操作。对于数千元素的数组，这种实现的执行速度会比原思路快几个数量级。

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