每日一题之旋转数组

题目

定义一个函数，实现数组的旋转。如输入 [1, 2, 3, 4, 5, 6, 7] 和 key = 3， 输出 [5, 6, 7, 1, 2, 3, 4]

第一种解法

• 将 k 后面的所有元素拿出来作为 part1
• 将 k 前面的所有元素拿出来作为 part2
• 返回 part1.concat(part2)

function rotate(arr, k) {
  const len = arr.length;
  if (!k || len === 0) return arr;
  const step = Math.abs(k % len);

  const part1 = arr.slice(-step);
  const part2 = arr.slice(0, len - step);
  const part3 = part1.concat(part2);
  return part3;
}

第二种解法

• 将 k 后面的元素，挨个 pop 然后 unshift 到数组前面

function rotate2(arr, k) {
  const length = arr.length;
  if (!k || length === 0) return arr;
  const step = Math.abs(k % length);

  for (let i = 0; i < step; i++) {
    const n = arr.pop();
    if (n != null) {
      arr.unshift(n);
    }
  }
  return arr;
}

功能测试

const arr = [1, 2, 3, 4, 5, 6, 7];
const a1 = rotate(arr, 3);
console.log("a1 :>> ", a1);
// a1 :>>  [
//   5, 6, 7, 1,
//   2, 3, 4
// ]

const a2 = rotate2(arr, 3);
console.log("a2 :>> ", a2);
// a2 :>>  [
//   5, 6, 7, 1,
//   2, 3, 4
// ]

性能测试

const arr1 = [];
for (let i = 0; i < 10 * 10000; i++) {
  arr1.push(i);
}
console.time("rotate1");
rotate(arr1, 9 * 10000);
console.timeEnd("rotate1"); // 885ms O(n^2)

const arr2 = [];
for (let i = 0; i < 10 * 10000; i++) {
  arr2.push(i);
}
console.time("rotate2");
rotate2(arr2, 9 * 10000);
console.timeEnd("rotate2"); // 1ms O(1) unshift性能很差