ruby - 将数组中的元素 i 旋转到 j

Question

给定一个数组：

a = [1, 2, 3, 4, 5, 6]

我想在某个方向上i旋转j元素n。因此，例如：

i = 2
j = 3
n = 1

旋转a会产生：

new_a = [1, 2, 4, 3, 5, 6]

这就是我所拥有的：

def rotate_sub(a, i, j, n)
  return a[0...i] + a[i..j].rotate(n) + a[j+1..-1]
end

有一个更好的方法吗？由于没有边界检查，i或者j很可能超出数组的范围。

Answer 1

如果您愿意改变原始数组，您可以执行以下操作：

a[i..j] = a[i..j].rotate n

但我喜欢你已经拥有的功能性解决方案。

Answer 2

我不认为有什么神奇的方法，所以也许最简单的就是最好的：

def rotate_sub(a, i, j, n)
  a[0...i] + a[i..j].rotate(n) + a[j+1..-1] if i < j && j < a.size
end