JS/TS中高效向Float32Array每第4位插入元素1的实现方案

云上老周

2026-9-29

性能瓶颈根因

你当前的方案性能差核心来自两个问题：

Array.splice 每次插入元素都会触发后续所有元素的内存移位，整体时间复杂度是O(n²)，面对几十万级元素时效率会指数级下降
把Float32Array转普通JS数组的过程本身也有额外的性能开销，完全可以省略

最优实现方案

我们可以直接预先分配好目标大小的内存空间，通过批量复制+定点赋值的方式完成操作，时间复杂度只有O(n)，性能是原方案的数百倍：

function pushToEveryFourthAndConvertToBuffer(floatArray: Float32Array): Buffer {
  const inputLen = floatArray.length;
  // 计算需要插入的1的数量：每3个原元素插1个，不满3个的最后也插1个
  const insertCount = Math.ceil(inputLen / 3);
  // 直接创建目标长度的Float32Array，提前分配内存
  const output = new Float32Array(inputLen + insertCount);
  
  let inputIdx = 0;
  let outputIdx = 0;
  while (inputIdx < inputLen) {
    // 批量复制最多3个原数组元素到目标位置
    const copyCount = Math.min(3, inputLen - inputIdx);
    output.set(floatArray.subarray(inputIdx, inputIdx + copyCount), outputIdx);
    outputIdx += copyCount;
    inputIdx += copyCount;
    // 插入固定值1
    output[outputIdx] = 1;
    outputIdx += 1;
  }
  // 直接从Float32Array转Buffer，不需要经过普通数组
  return Buffer.from(output.buffer);
}