[Easy] 手写 chunk

题目描述

Lodash 的 chunk 是开发中经常被用的效用函式，也经常会在面试被问到。 chunk 能将输入的数组分割成指定长度的小组。并输出一个新的数组，包含原始数组的元素，并按指定的 size 分组成子数组。

const originalArray = [1, 2, 3, 4, 5, 6, 7, 8, 9];
const chunkSize = 3;

// 输出: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
const chunkedArray = chunkArray(originalArray, chunkSize);

本题解答

以下是本题的解答，详细解题思路可以在 E+ 成长计划看到

解法一

function chunk(array, size = 1) {
  let temp = [];
  const result = [];
  for (let i = 0; i < array.length; i++) {
    temp.push(array[i]);
    if (temp.length === size || i === array.length - 1) {
      result.push(temp);
      temp = [];
    }
  }
  return result;
}

解法二

function chunk(array, size) {
  const chunkedArray = [];

  for (let i = 0; i < array.length; i += size) {
    chunkedArray.push(array.slice(i, i + size));
  }

  return chunkedArray;
}