给定一个索引，相应地交换数组元素

Question

我想要实现的有点像无线电组行为，一次只能选择一个无线电输入。 例如，起始数组[0, 1, 1, 1, 1, 1]应按以下方式交换其元素：

指数	大批
0	[0, 1, 1, 1, 1, 1]
1	[1, 0, 1, 1, 1, 1]
2	[1, 1, 0, 1, 1, 1]
3	[1, 1, 1, 0, 1, 1]
4	[1, 1, 1, 1, 0, 1]
5	[1, 1, 1, 1, 1, 0]

我想出了这个，但我认为它在某些情况下会“额外工作”（不必要的循环）。

function rearrange(array: number[], idx: number) {
  let arr = array.slice();
  let l = arr.length;
  if (arr.indexOf(0) === idx) return arr;

  while (arr.indexOf(0) !== idx) {
    let swap;
    for (let i = 0; i < l; i++) {
      if (arr[i] === 0 || arr[i + 1] === 0) {
        swap = arr[i];
        if (i + 1 < l) {
          arr[i] = arr[i + 1];
          arr[i + 1] = swap;
        }
        if (i + 1 > l) {
          arr[i] = arr[i - 1];
          arr[i - 1] = swap;
        }
      }
    }
  }

  return arr;
}

我想知道您是否对如何使这个过程更简单/更好有想法。

Answer 1

只需用findIndex识别前一个 0 ，为其分配 1 ，然后将 0 分配给idx ？

function rearrange(array: number[], idx: number) {
  const arr = [...array];
  arr[arr.indexOf(0)] = 1;
  arr[idx] = 0;
  return arr;
}

另一种方法...

const rearrange = (array: number[], idx: number) => (
  array.map((_, i) => i === idx ? 0 : 1)
);

Answer 2

您可以使用切片和拼接来“一次性”旋转子数组，而不是交换对。

这是一个 function 完成这项工作，改变给定数组（不返回新数组），以及一系列调用来演示它。 我假设值可能会变化，并且需要保持在一起，就好像它们与“移动”0交换一样：

 function rearrange(array, idx) { let zero = array.indexOf(0); if (zero < idx) { arr.splice(zero, idx - zero + 1, ...arr.slice(zero + 1, idx + 1), 0); } else { arr.splice(idx, zero - idx + 1, 0, ...arr.slice(idx, zero)); } } arr = [0,1,2,3,4,5]; rearrange(arr, 3); console.log(...arr); rearrange(arr, 1); console.log(...arr); rearrange(arr, 5); console.log(...arr); rearrange(arr, 1); console.log(...arr);