Как добавить массив id потомков по id родителей?

Question

Владимир Допустим @DopustimVladimir

Веб-ориентированный программист

JavaScript

Как добавить массив id потомков по id родителей?

Всем привет! Допустим, существует вот такая коллекция:

var cats = [
{ id: 1, parent_id: 0, title: 'Кино' },
{ id: 2, parent_id: 0, title: 'Музыка' },
{ id: 3, parent_id: 1, title: 'Зарубежное кино' },
{ id: 4, parent_id: 2, title: 'Зарубежная музыка' },
{ id: 5, parent_id: 3, title: 'Новое' },
{ id: 6, parent_id: 3, title: 'Топ10' },
{ id: 7, parent_id: 4, title: 'Новое' },
{ id: 8, parent_id: 4, title: 'Топ10' },
{ id: 9, parent_id: 0, title: 'Софт' }
];

Нужно сделать так:

var catsExtended = [
{ id: 1, parent_id: 0, childs: [ 3, 5, 6 ], title: 'Кино' },
{ id: 2, parent_id: 0, childs: [ 4, 7, 8 ], title: 'Музыка' },
{ id: 3, parent_id: 1, childs: [ 5, 6 ], title: 'Зарубежное кино' },
{ id: 4, parent_id: 2, childs: [ 7, 8 ], title: 'Зарубежная музыка' },
{ id: 5, parent_id: 3, childs: [], title: 'Новое' },
{ id: 6, parent_id: 3, childs: [], title: 'Топ10' },
{ id: 7, parent_id: 4, childs: [], title: 'Новое' },
{ id: 8, parent_id: 4, childs: [], title: 'Топ10' },
{ id: 9, parent_id: 0, childs: [], title: 'Софт' }
];

Вопрос задан более трёх лет назад
294 просмотра

Комментировать

Подписаться 1 Оценить Комментировать

Answer 1 · 2016-07-05 10:08:25

1) преобразуете массив в объект с помощью _.indexBy из lodash/underscore при этом так как внутри массива объекты, то полученый объект будет содержать ссылки на те же объекты
2) пробегаем исходный массив и отмечаем в childs у родителя

UPD: писал с телефона в дороге, дополню кодом
Простой вариант, как описано мною выше:

var catsIndex = _.indexBy(cats, 'id');
cats.forEach(function(el) {
  if(!catsIndex[el.parent_id]) return;
  if(!catsIndex[el.parent_id].childs) catsIndex[el.parent_id].childs = [];
  catsIndex[el.parent_id].childs.push(el.id);
});
console.log(cats);

Оптимизированный вариант, в один проход:

var catsChilds = {};
for(var i = cats.length; i--;) {
  if(catsChilds[cats[i].id]) {
    cats[i].childs = catsChilds[cats[i].id];
  } else {
    cats[i].childs = catsChilds[cats[i].id] = [];
  }
  if(!catsChilds[cats[i].parent_id]) {
    catsChilds[cats[i].parent_id] = [];
  }
  catsChilds[cats[i].parent_id].push(cats[i].id);
}
console.log(cats);

P.S. если нужно именно новый массив, не затрагивая исходный, то перед выполнением кода нужно сделать deapClone и работать уже с клоном, реализацию deapClone можете легко нагуглить

Как добавить массив id потомков по id родителей?

Войдите на сайт