Суммируйте дочерние элементы в несколько значений и сохраняйте результат в родительский элемент в n-арном дереве в Javascript

1

Я спросил: суммирует значения детей и сохраняет результат для родителя в n-арном дереве в Javascript

И я получил ответ от @Nina Scholz. Теперь я пытаюсь использовать тот же код, но с более чем одним значением, например:

value: {A: 1, B: 2, C: 3}

Я попробовал что-то вроде этого:

function update(array) {
   var root = [],
      references = array.reduce((r, o) => {
         if (!o.parents.length) {
            root.push(o.id);
         }
         r[o.id] = o;
         return r;
      },Object.create(null));

   root.reduce(
   function sum(s, id) {
     var o = references[id];

     for (var key in o.value) {
       return s + (o.value[key] = o.children.reduce(sum, 0) || o.value[key])
     }
   }, 0);

   return array;
}

var data2 = [{ id: 'A', parents: [], children: ['B', 'E'], value: {A:1, B:2} }, { id: 'B', parents: ['A'], children: ['C', 'D'], value: {A:1, B:2} }, { id: 'C', parents: ['B'], children: [], value: {A:1, B:2} }, { id: 'D', parents: ['B'], children: [], value: {A:1, B:2} }, { id: 'E', parents: ['A'], children: ['F'], value: {A:1, B:2} }, { id: 'F', parents: ['E'], children: [], value: {A:1, B:2} }]

console.log(update(data2));

Но работает только со значением A из-за возврата.

Еще одна вещь, которую я пробовал:

function update(array) {
   var root = [],
      references = array.reduce((r, o) => {
         if (!o.parents.length) {
            root.push(o.id);
         }
         r[o.id] = o;
         return r;
      },Object.create(null));

   root.reduce(
   function sum(s, id) {
     var o = references[id];

     var x;
     for (var key in o.value) {
        x = (s + (o.value[key] = o.children.reduce(sum, 0) || o.value[key]));
     }
   }, 0);

   return array;
}

var data2 = [{ id: 'A', parents: [], children: ['B', 'E'], value: {A:1, B:2} }, { id: 'B', parents: ['A'], children: ['C', 'D'], value: {A:1, B:2} }, { id: 'C', parents: ['B'], children: [], value: {A:1, B:2} }, { id: 'D', parents: ['B'], children: [], value: {A:1, B:2} }, { id: 'E', parents: ['A'], children: ['F'], value: {A:1, B:2} }, { id: 'F', parents: ['E'], children: [], value: {A:1, B:2} }]

console.log(update(data2));

Но это не работает с теми, у кого есть дети.

  • 0
    Какой у Вас вопрос?
Теги:
arrays
recursion

1 ответ

2
Лучший ответ

Вы можете адаптировать sum и итерацию ключа/значения для добавления к объекту value.

function update(array) {

    function sum(s, id) {
        var o = references[id];
        o.value = o.children.reduce(sum, undefined) || o.value;
        if (!s) {
            return Object.assign({}, o.value);
        }
        Object
            .entries(o.value)
            .forEach(([k, v]) => s[k] = (s[k] || 0) + v);
        return s;
    }

    var root = [],
        references = array.reduce((r, o) => {
            if (!o.parents.length) {
                root.push(o.id);
            }
            r[o.id] = o;
            return r;
        }, Object.create(null));

    root.reduce(sum, undefined);

    return array;
}

var data2 = [{ id: 'A', parents: [], children: ['B', 'E'], value: { A: 1, B: 2 } }, { id: 'B', parents: ['A'], children: ['C', 'D'], value: { A: 1, B: 2 } }, { id: 'C', parents: ['B'], children: [], value: { A: 1, B: 2 } }, { id: 'D', parents: ['B'], children: [], value: { A: 1, B: 2 } }, { id: 'E', parents: ['A'], children: ['F'], value: { A: 1, B: 2 } }, { id: 'F', parents: ['E'], children: [], value: { A: 1, B: 2 } }]
console.log(update(data2));
.as-console-wrapper { max-height: 100% !important; top: 0; }

Ещё вопросы

Сообщество Overcoder
Наверх
Меню