[Medium] LeetCode JS 30 - 2631. Group By (阵列分组)

题目描述

请实践一个阵列操作的方法 groupBy，该方法会让任何阵列呼叫 array.groupBy(fn) 时，获得该阵列分组后的结果。

所谓的阵列分组**，**是指阵列中的每个键都是 fn(arr[i]) 的输出，输出的阵列中，含有原始阵列中具有该键的所有项。

提供的回呼函式 fn 会接受阵列中的项目并传回一个字串类型的键。每个值阵列的顺序，会与阵列中元素出现的顺序相同。任何顺序的键都是可以接受的。

具体如下范例，输入一个有物件的阵列，并且回呼函式是以 item.id 作为判断，所以呼叫完 groupBy(fn) 后，会把 id 做为键，而有该 id 的物件，都会被放到同个阵列做为值，例如有 {"id":"1"} 的会被放在同个阵列。

// 范例
输入：
[
  {"id":"1"},
  {"id":"1"},
  {"id": 2},
],
fn = function (item) {
  return item.id;
}

输出：
{
  "1": [{"id": "1"}, {"id": "1"}],
  “2”：[{“id”：“2”}]
}

解释：
输出来自函式 array.groupBy(fn)。
分组方法是从阵列中的每个项目中取得 "id"。
有两个 "id" 为 1 的物件。所以将这两个物件都放在第一个阵列中。
有一个 "id" 为 2 的物件。所以该物件被放到第二个阵列中。

本题解答

以下是本题的解答，详细解题思路可以在 E+ 成长计划看到。如果想练习更多题目，推荐可以到 GreatFrontEnd 上练习

解法

Array.prototype.groupBy = function (fn) {
  const grouped = {};

  this.forEach((item) => {
    const key = fn(item);
    if (!grouped[key]) {
      grouped[key] = [];
    }
    grouped[key].push(item);
  });
  return grouped;
};