整合对象数组的算法问题，求解！

Question

源数据

const resData = [
  {dict: '东北', type: '二线', isProvinceCapital: '是', city: '哈尔滨', money: 100},
  {dict: '东北', type: '二线', isProvinceCapital: '是', city: '沈阳', money: 200},
  {dict: '东北', type: '三线', isProvinceCapital: '否', city: '大兴安岭', money: 300},
  {dict: '华东', type: '一线', isProvinceCapital: '是', city: '上海', money: 400},
  {dict: '华东', type: '二线', isProvinceCapital: '是', city: '杭州', money: 500},
  {dict: '华东', type: '二线', isProvinceCapital: '否', city: '宁波', money: 600}
];
// 处理数据的方法!
function formatData(keys) {

}

下面是输入和期望输出结果

// 1
// 输入 ['isProvinceCapital']
// 输出如下：
let list = [
  {
    isProvinceCapital: '是',
    children: [
      {dict: '东北', type: '二线', isProvinceCapital: '是', city: '哈尔滨', money: 100},
      {dict: '东北', type: '二线', isProvinceCapital: '是', city: '沈阳', money: 200},
      {dict: '华东', type: '一线', isProvinceCapital: '是', city: '上海', money: 400},
      {dict: '华东', type: '二线', isProvinceCapital: '是', city: '杭州', money: 500},
    ]
  },
  {
    isProvinceCapital: '否',
    children: [
      {dict: '东北', type: '三线', isProvinceCapital: '否', city: '大兴安岭', money: 300},
      {dict: '华东', type: '二线', isProvinceCapital: '否', city: '宁波', money: 600}
    ]
  }
];

// 2
// 输入 ['dict', 'type']
// 输出如下：
let list2 = [
  {
    dict: '东北',
    children: [
      {
        dict: '东北',
        type: '二线',
        children: [
          {dict: '东北', type: '二线', isProvinceCapital: '是', city: '哈尔滨', money: 100},
          {dict: '东北', type: '二线', isProvinceCapital: '是', city: '沈阳', money: 200}
        ]
      },
      {
        dict: '东北',
        type: '三线',
        children: [
          {dict: '东北', type: '三线', isProvinceCapital: '否', city: '大兴安岭', money: 300}
        ]
      }
    ]
  },
  {
    dict: '华东',
    children: [
      {
        dict: '华东',
        type: '一线',
        children: [
          {dict: '华东', type: '一线', isProvinceCapital: '是', city: '上海', money: 400}
        ]
      },
      {
        dict: '华东',
        type: '二线',
        children: [
          {dict: '华东', type: '二线', isProvinceCapital: '是', city: '杭州', money: 500},
          {dict: '华东', type: '二线', isProvinceCapital: '否', city: '宁波', money: 600}
        ]
      },
    ]
  }
];

求formatData方法如何实现，感谢

Answer 1

const formatData = (data, keys) => Object.entries(
    data.reduce((acc, item) => (
        (acc[item[keys[0]]] ??= []).push(item),
        acc
    ), {})
).map(([ keyvalue, children ]) => ({
    [keys[0]]: keyvalue,
    children: keys.length > 1
        ? formatData(children, keys.slice(1)).map(child => ({
            [keys[0]]: keyvalue,
            ...child
        }))
        : children
}))

这个只带了本层的 key，我睡觉了明天完善一下
好啦
又修了一下，确实好啦

Answer 2

就是递归多级分组。分组可以用 .reduce 自己实现，我懒，用 Lodash

import _ from "lodash";

function formatData(data, keys) {
    if (!(keys?.length)) { return data; }
    const [key, ...restKeys] = keys;
    return _(resData)
        .groupBy(it => it[key])
        .entries()
        .map(([name, children]) => ({
            [key]: name,
            children: formatData(children, restKeys)
        }))
        .value();
}