Как перестроить массив объектов?

Question

[Дмитрий Гололобов]

Всем привет!

Есть массив категорий:

[
{id: 1, name: 'Категория 0-1', parent: 0},
{id: 2, name: 'Категория 0-2', parent: 0},
{id: 3, name: 'Категория 0-3', parent: 0},
{id: 4, name: 'Категория 1-1', parent: 1},
{id: 5, name: 'Категория 1-2', parent: 1},
]

на выходе хочу получить:

[
{id: 1, name: 'Категория 0-1', parent: 0,
 child: [
 {id: 4, name: 'Категория 1-1', parent: 1},
 {id: 5, name: 'Категория 1-2', parent: 1},
]
},
{id: 2, name: 'Категория 0-2', parent: 0},
{id: 3, name: 'Категория 0-3', parent: 0},
]

Уровней вложенности может быть сколько угодно.

Помогите составить рекурсивную функцию.

Answer 1

[Антон Швец]

const tree = arr.reduce((sum, el) => {
	if (el.parent === 0) {
  	sum.push(el)
  } else {
  	const parent = arr.find(p => p.id === el.parent);
  	parent.children = parent.children || [];
  	parent.children.push(el);
  }
  return sum;
}, []);

тут нет нужды в рекурсии.
Следует отметить, что элементы исходного массива тоже будут изменены. Если это неприемлемо, нужно сначала склонировать.
O(n^2)

А вот обратная задача будет уже решаться с помощью рекурсии.

Comments

[Дмитрий Гололобов]

В данном варианте все помещается в первую категорию. Не подходит

[Антон Швец]

Дмитрий Гололобов, да, я там присваивание вместо равно поставил.
Поправил
https://jsfiddle.net/melchiorio/8ydx0cz2/2/

[Дмитрий Гололобов]

Антон, ну а моя лень не позволила мне внимательно просмотреть функцию и я признал ее не удачной ))
Все ок! Работает!

Answer 2

[Александр]

Нужна рекурсивная функция:

var data = [
{id: 1, name: 'Категория 0-1', parent: 0},
{id: 2, name: 'Категория 0-2', parent: 0},
{id: 3, name: 'Категория 0-3', parent: 0},
{id: 4, name: 'Категория 1-1', parent: 1},
{id: 5, name: 'Категория 1-2', parent: 1},
];
var getChilds = (data,par)=>data.filter((el)=>el.parent == par).map(el=>{el["childs"]=getChilds(data,el.id);return el;});
var tree = getChilds(data,0)

результат JSON.stringify(tree,"\n",4)

[
    {
        "id": 1,
        "name": "Категория 0-1",
        "parent": 0,
        "childs": [
            {
                "id": 4,
                "name": "Категория 1-1",
                "parent": 1,
                "childs": []
            },
            {
                "id": 5,
                "name": "Категория 1-2",
                "parent": 1,
                "childs": []
            }
        ]
    },
    {
        "id": 2,
        "name": "Категория 0-2",
        "parent": 0,
        "childs": []
    },
    {
        "id": 3,
        "name": "Категория 0-3",
        "parent": 0,
        "childs": []
    }
]

Comments

[Дмитрий Гололобов]

Вот! Спасибо, а я вокруг да около хожу, главное вывод списка ul>li>ul>li.... из изначального массива сделал с помощью рекурсии, а переделать в json его не смог )

Answer 3

[Rsa97]

Если в исходном массиве потомки всегда идут после родителя, то никакой рекурсии не надо.

var flat = [
  {id: 1, name: 'Категория 0-1', parent: 0},
  {id: 2, name: 'Категория 0-2', parent: 0},
  {id: 3, name: 'Категория 0-3', parent: 0},
  {id: 4, name: 'Категория 1-1', parent: 1},
  {id: 5, name: 'Категория 1-2', parent: 1}
];

var els = [];
var tree = [];
for (var i = 0; i < flat.length; i++) {
  flat[i].childs = [];
  if (flat[i].parent == 0) {
    tree.push(flat[i]);
  } else {
    els[flat[i].parent].childs.push(flat[i]);
  }
  els[flat[i].id] = flat[i];
}
console.log(tree);

Comments

[Дмитрий Гололобов]

В том-то и дело, что не всегда. Но вариант имеет право на жизнь.
Я для упрощения такой массив предоставил.