Ruby 变量在函数中无法正常工作

Question

我有这个功能：

def find_deleted_number(arr, mixed_arr)

    default = 0

    arr.each do |i|
        if mixed_arr.include?(arr[i]) == true
            default += arr[i]
        else 
            return arr[i]
        end
    end
    return default  
end

find_deleted_number([1,2,3,4,5], [3,4,1,5])
#=> 2 (as expected)

find_deleted_number([1,2,3,4,5,6,7,8,9], [1,9,7,4,6,2,3,8])
#=> 5 (as expected)

find_deleted_number([1,2,3,4,5,6,7,8,9], [5,7,6,9,4,8,1,2,3])
#=> nil (expected 45)

这个想法是让函数检查哪个数字在 arr 但不在 mixed_arr 中。这一切都很好，但问题是当两个数组相同时，我希望它给我其中一个数组中每个值的总和（这就是我做 default += arr[i] 的原因）。但是结果是nil，我不太明白为什么。如果可能的话，我想了解它失败的原因。

Answer 1

arr.each 迭代 arr 的元素，而不是索引。因此，您需要将 arr[i] 更改为 i.

完整答案是：

def find_deleted_number(arr, mixed_arr)
  default = 0

  arr.each do |i|
    if mixed_arr.include?(i)
      default += i
    else 
      return i
    end
  end

  default  
end

Answer 2

如@yzalavin 所说，它失败的原因。

您还可以简化代码：

def find_deleted_number(arr, mixed_arr)
  return arr.sum if arr.sort == mixed_arr.sort

  (arr - mixed_arr).first
end

@Stefan 提供了更短的单行代码：

(arr - mixed_arr).first || arr.sum

Ruby 变量在函数中无法正常工作

Ruby variable not working properly in function

ruby

loops

each