分组

74 阅读2分钟

前言:锻炼自己的思想,规范自己的编程思路。每天一道算法题,督促自己。

靡不有初,鲜克有终,加油坚持下去。

问题:请你编写一段可应用于所有数组的代码,使任何数组调用 array. groupBy(fn) 方法时,它返回对该数组 分组后 的结果。

数组 分组 是一个对象,其中的每个键都是 fn(arr[i]) 的输出的一个数组,该数组中含有原数组中具有该键的所有项。

提供的回调函数 fn 将接受数组中的项并返回一个字符串类型的键。

每个值列表的顺序应该与元素在数组中出现的顺序相同。任何顺序的键都是可以接受的。

请在不使用 lodash 的 _.groupBy 函数的前提下解决这个问题。

示例:

输入:
array = [
  {"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 的对象。所以该对象被放到第二个数组中。

思路:这个题目看起来还是蛮复杂的,让我有点稀里糊涂的,中等题,我接触的第一个JavaScript的中等题,果然有点难度。

传入一个数组和一个fn函数,确定分组的函数,我想到肯定用for循环遍历,将数组中每个值放入fn函数,比较后传入新的分组数组中。

我先定义一个对象并返回,然后循环遍历数组中的每一个元素,调用fn(item)来获取对应的key。如果key不在对象上,就初始化一个数组,如果在对象上则添加元素。

基于上述思考,代码如下:

Array.prototype.groupBy = function(fn) {
    let ans = {};
  for (let item of this) {
    let key = fn(item);
    if (ans[key]) {
      ans[key].push(item);
    } else {
      ans[key] = [item];
    }
  }
  return ans;
};

执行结果如下图:

image-20230504194109768.png

结论:这个题目对我来说,理解不难,写起来还挺难的,在看过评论区后,才得出解决的方法。