Ruby:检查所有数组元素是否相等

Question

我对Ruby代码有点'问题'.我想检查一个数组的所有元素是否相等.

例如,假设我有一个只有5的数组:

arr = [5, 5, 5, 5, 5]

我知道我可以做点什么

arr[0] == arr[1] == arr[2] == arr[3] # == arr[4] == ...

但是对于大型阵列来说这是不可能的,而且在我看来也不像Ruby那样.我们可以通过这样做来改进它:

def all_equal?(arr)
  for i in 0..(arr.size-2)
    if arr[i] != arr[i+1] then
      return false
    end
  end
  true
end

但我也认为这很难看.那么有没有内置/更好/更短(更多Ruby风格)的方式来做到这一点？

TL; DR什么是检查数组是否只包含一个不同元素(例如 [5, 5, 5])的最短/最Ruby特有的方法？

谢谢.

Answer 1

你也可以使用.uniq,返回一个没有重复的数组,并检查大小:

def all_equal?(arr)
    arr.uniq.size <= 1
end

Answer 2

几种方式.

最好的:

array.uniq.count <= 1 # or == 1 if it can't be an empty array

和:

array == ([array.first] * array.count)

和:

(array | array).count <= 1 # basically doing the same thing as uniq

也:

array.reduce(:|) == array.first # but not very safe

如果它是一个可排序的数组,那么:

array.min == array.max    

而且,只是为了多样化:

!array.any?{ |element| element != array[0] } # or array.first instead of array[0]

或者:

array.all?{ |element| element == array[0] } # or array.first instead of array[0]

Answer 3

使用Enumerable#each_cons：

def all_equal?(xs)
  xs.each_cons(2).all? { |x, y| x == y }
end