Ruby:创建数组项 "on demand"

Question

在带哈希的 Ruby 中，我可以这样做:

h = Hash.new { |h,k| h[k] = "created #{k}" }

因此，每次我尝试使用散列中不存在的键访问项目时，它都会调用该 block 并创建这个新项目并使用该键存储。

是否有类似的方法来处理数组？

Answer 1

Array.new 方法可以接收一个 block 。它传递元素的索引， block 的结果存储在数组中。

Array.new(3) { |index| index ** 2 }
# => [0, 1, 4]

但是，所有元素都将在您调用该方法时创建。它们也将存储在 block 中，无法阻止。

我们可以将 Array 子类化并实现所需的类似 Hash 的行为。

class CustomArray < Array
  def [](index)
    if super.nil? then @default_proc.call self, index end
    super
  end
end

class << CustomArray
  def new(*arguments, **keyword_arguments, &block)
    if arguments.empty? and block
      super().tap do |array|
        array.instance_variable_set :@default_proc, block
      end
    else super end
  end
end

这样，通常的 Array API 就被保留了下来。如果大小参数未随 block 一起传递，它将用作默认的 Proc。

array = CustomArray.new { |array, index| array[index] = index + 2 }

p array[10]
# => 12

p array
# => [nil, nil, nil, nil, nil, nil, nil, nil, nil, nil, 12]

看它运行here .

请注意，由于 nil 的含义，此实现存在问题。在这种情况下，由于数组的工作方式，它被定义为 empty 值，这在连续内存的上下文中是合理的。如果您将元素存储在大于数组大小的索引处，它将用 nil 填充空白，以指示最后一个元素和您刚刚插入的元素之间的空白。

如果 nil 是一个有意义的值，请考虑使用带有整数键的 Hash 而不是 Array。