arrays - 从保留索引的数组中删除一个项目

Question

有一个对象数组，其中很多对象在数组中出现了多次。有一个索引到一个项目的特定位置。我想从数组的所有位置删除一个对象，同时保持索引。已删除项的索引应移动到下一个可用项，如果没有，则它应回绕到数组的开头。

我觉得我解释得不是很好，所以这里有一个简单的例子:

array = [:b, :a, :b]
index = 2

现在从数组中删除 :a，索引现在应该为 1，因为第二个 :b 现在位于索引 1。

假设一个索引指向已删除的对象，它应该移动到下一个可用的对象。因此，如果索引为 1，则它不会更改，因为它现在指向紧跟在 :a 之后的 :b。

这是一个环绕的例子:

array = [:b, :a]
index = 1

如果您删除 :a，那么索引应该环绕到下一个可用索引，或者 0。

看似简单，但有更复杂的边缘情况。这是我能想到的最复杂的示例:

array = [:a, :a, :b, :a, :c, :b, :a]

删除':a'后，结果数组为[:b, :c, :b]。从所有可能的起始索引(0 到 6)开始，它们应更改为 0、0、0、1、1、2、0。

假设数组中有多个对象，删除后它永远不会变空。

(编辑)

感谢@Raffael 告诉我应该包括我目前为止尝试过的内容。这是我的第一次尝试:

• @items 是数组。
• @current_item_index为需要调整的索引。
• item 是要删除的项目。

代码:

indexes = @items.each_index.each_with_object([]) do |i, memo|
  memo << i if @items[i] == item
end

indexes.reverse.each do |i|
  @items.delete_at(i)
  @current_item_index -= 1 if i < @current_item_index
  @current_item_index = 0 unless @current_item_index < @items.size
end

它看起来不是很“Ruby”(不清晰也不简洁)。性能似乎也不是很好，因为它在列表中运行了两次——更不用说所有那些不太好的 delete_at 调用了。

Answer 1

新索引等于旧数组中元素的个数

• 放在旧索引的前面
• 将保留在数组中

您可以像这样在构建新数组时随时计算这些数:

def delete_and_follow(old_array, old_index, unwanted)
  new_array, new_index = [], 0
  old_array.each.with_index do |item, idx|
    unless item == unwanted
      new_array << item
      new_index += 1 if idx < old_index
    end
  end
  new_index = 0 if new_index >= new_array.length
  [new_array, new_index]
end

或者用更少的代码做同样的事情(虽然有点神秘):

def delete_and_follow(old_array, old_index, unwanted)
  new_index = old_array.take(old_index).count{ |item| item != unwanted }
  new_array = old_array - [unwanted]
  [new_array, new_index % new_array.length]
end

喂!