如何用Ruby实现数组元素单步上移/下移？及相邻元素交换方法

嘿，这个需求挺常见的，我来给你拆解一下怎么做～

先解决你给出的具体交换案例 #

你有数组 array = ['a', 'b', 'c', 'd']，想要交换'b'和'c'得到['a', 'c', 'b', 'd']，步骤很简单：

1. 先定位'b'的索引：

array = ['a', 'b', 'c', 'd']
idx = array.index('b') # 得到 idx = 1

2. 用Ruby的并行赋值直接交换它和下一个元素的位置：

array[idx], array[idx+1] = array[idx+1], array[idx]

3. 现在你打印array就会得到：['a', 'c', 'b', 'd']

这种方法的好处是不需要额外的临时变量，Ruby会先把右边两个元素的值都缓存好，再赋值给左边，避免了中间值被覆盖的问题。

通用的元素单步上移/下移方法 #

如果需要更通用的工具，比如让任意元素单步上移或下移（还要处理边界情况，比如元素已经在数组首尾时不做操作），可以写两个小方法：

1. 元素单步下移（和下一个元素交换） #

def move_down(array, target_element)
  idx = array.index(target_element)
  # 处理边界：元素不存在，或者已经在最后一位，直接返回原数组副本
  return array.dup if idx.nil? || idx == array.length - 1

  # 交换位置
  array[idx], array[idx+1] = array[idx+1], array[idx]
  array
end

# 使用示例：
original_array = ['a', 'b', 'c', 'd']
new_array = move_down(original_array, 'b')
puts new_array.inspect # 输出: ["a", "c", "b", "d"]

2. 元素单步上移（和前一个元素交换） #

def move_up(array, target_element)
  idx = array.index(target_element)
  # 处理边界：元素不存在，或者已经在第一位，直接返回原数组副本
  return array.dup if idx.nil? || idx == 0

  # 交换位置
  array[idx], array[idx-1] = array[idx-1], array[idx]
  array
end

# 使用示例：
original_array = ['a', 'b', 'c', 'd']
new_array = move_up(original_array, 'c')
puts new_array.inspect # 输出: ["a", "c", "b", "d"]

小提示 #

• 上面的方法里用了array.dup返回原数组的副本，避免修改原始数组；如果需要直接修改原数组，去掉.dup即可。
• 如果你的数组里有重复元素，array.index(target_element)只会返回第一个匹配元素的索引。如果要处理重复元素的情况，需要遍历数组找到所有目标索引再分别操作。

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