求同级叶子数据总和的乘积

在树形数据结构中，有时需要对同级叶子节点的数据进行总和并相乘。本文将介绍一个解决方案。

数据结构

首先定义一个树形数据结构，包含 value 和 children 两个属性。

class TreeNode {
  constructor(value) {
    this.value = value;
    this.children = [];
  }
}

求同级叶子数据总和

首先定义一个递归函数 sumLeaf，用于计算同级叶子节点的数据总和。函数的参数包括树节点和同级叶子节点的数据和，返回同级叶子节点的数据和。

function sumLeaf(node, sum) {
  if (!node.children.length) {  // 如果节点无子节点，则为叶子节点
    return sum + node.value;    // 将叶子节点的数据加入总和并返回
  }
  let total = 0;
  for (let child of node.children) {
    total += sumLeaf(child, sum);  // 递归调用子节点
  }
  return total;
}

求同级叶子数据总和的乘积

接下来定义一个函数 multiplyLeafTotal，用于计算同级叶子节点的数据总和并相乘。函数的参数为树节点，返回同级叶子节点的数据总和的乘积。

function multiplyLeafTotal(node) {
  let total = sumLeaf(node, 0);  // 计算同级叶子数据的总和
  let siblings = node.parent ? node.parent.children.filter((child) => child !== node) : [node];  // 获取同级节点
  let siblingTotal = 0;
  for (let sibling of siblings) {
    siblingTotal += sumLeaf(sibling, 0);  // 计算同级叶子数据的总和并相加
  }
  let result = total * siblingTotal;  // 乘积
  return result;
}

使用示例

下面是一个使用示例，构建了一棵树形数据结构并计算同级叶子节点的数据总和的乘积。

let root = new TreeNode(1);
let node1 = new TreeNode(2);
let node2 = new TreeNode(3);
let node3 = new TreeNode(4);
let node4 = new TreeNode(5);
let node5 = new TreeNode(6);
root.children = [node1, node2, node3];
node2.children = [node4, node5];

console.log(multiplyLeafTotal(node4));  // 输出 360

总结

本文介绍了一个计算同级叶子节点的数据总和并相乘的解决方案。通过递归计算同级叶子数据的总和，并通过过滤同级节点来计算同级叶子数据的总和的乘积。