Javascript：当用户从树中选择/取消选择节点时如何保留树结构和祖父母关系

Question

我有一个树形结构{name, [children]}（JS 对象）。用户可以任意选择任何节点以部分复制结构（复制到 JS 对象数组中，因为可以选择松散叶节点）。例如用户可以选择一个叶子节点，但不能选择它的父/祖父节点，那么这个叶子节点只会在最高父节点的children 的平面数组中。

例子：

original (JS Object)      selected
   A✓                       A
   |  \ \                  / \
   B✓ C D✓                B  D
  / \          =>         / \
 E✓  F                   E  G
     |
     G✓ 

original                  selected(Array of JS objects)
   A                       B, D      
   |  \ \                 / \
   B✓ C D✓               E  G 
  / \          =>         
 E✓  F                   
     |
     G✓ 

我的第一个想法是：对于点击的每个节点，假设我们维护一个 selected 数组：

1. 如果这个节点在selected => 取消选择（相当直接）：

   a1。如果节点是“顶级”元素 => 将其从数组中删除，并将其子节点推送到数组中。

   a2。如果节点是顶级元素的（大）子节点 => 从其父节点的子节点中删除该节点，并将该节点的子节点重新附加到其父节点。

2. 如果这个节点不在selected => 选择（奇怪的事情发生）：

   一个。用空的children 初始化tempNode，因为我们还不知道它的孩子

   b.如果这个节点有 (grand)children => 对于每个 (grand)child，如果 (grand)child 在 selected => 删除 selected 中的 (grand)child => 将这个 (grand)child 添加到tempNode's children（应该递归多级）

   c1.如果此节点的（祖）父级已经在 selected 中 => 将 tempNode 附加到（祖）父级（应该是多个级别的递归）

   c2。如果该节点的（大）父节点不在selected 中，则将其推送到数组中。

我主要停留在步骤 2b 和 2c。

//for adding node into selected
const addChildInSelected = (child) => {
    var tempNode = {
        name: child.name,
        children: []
    }
    var clonedState = clone(selected)
    var childInTree = findChildInTree(child.id, props.data)
    //if the child has children, check if they are in selected yet.
    if (childInTree.children.length > 0) {
        //Problem 1: I should run this recursively for all the grandchildren, 
        //but I struggle to find the base case. But recursion might not be the best solution
        for (var i = 0; i < childInTree.children.length; i++) {
            //Problem 2: findChildInSelected/removeChildInSelect are all recursive calls as well,
            //due to the tree structure. very worried about stack overflow during this step...
            var grandChildInSelected = findChildInSelected(childInTree.children[i].id)
            if (grandChildInSelected !== null) {
                clonedState = removeChildInSelected(clonedState)
                tempNode.children.push(findChildInTree(childInTree.children[i]))
            }
        }
    }
    //Problem 3. I realized I needed to check for each node in the `selected` again. 
    //another potential performance hurdle
}

现在我重新考虑这个问题，每个节点只关心它自己、它的直接父/祖父以及它的所有子/孙作为一个平面数组。这可能是研究这个问题的一个很好的视角。也许找到父母/祖父母或所有孩子的辅助功能将是一个好的开始。 （另外我意识到在几个函数中我忘记检查祖父母......）

但由于我使用多个树结构进行操作，所以下面的函数似乎并不理想......

//just a quick example
const findDescendantsInSelected = (node, descendants=[]) => {
    //recursive call within recursive function which could overflow if the tree is big
    if (findNodeInSelected(node) !== null) {
        descendents.push(node)
    }
    if (node.children.length > 0) {
        for (var i = 0; i < entry.children.length; i++) {
            result = findDescendantsInSelected(node.children[i], descendants);
        }
        return result
    }
    return descendents
}

总而言之，维护几乎 3 个独立的树（选择状态、原始数据和每个步骤的克隆，因为我使用了 react）并在每个步骤中跟踪父母/孩子，跨不同的功能让我很头疼。任何有关如何简化问题的见解将不胜感激！

Answer 1

使用深度优先遍历的简单纯递归函数就足够了。它返回您想要的选定树的列表（称为forest）。如果选择了一个节点，它将收集所有选定的后代树并返回一个以这些为子节点的新节点。如果该节点没有被选中，它只会加入来自其子节点的列表并返回该列表。

function selectedForest(node) {
    const trees = node.children.flatMap(selectedForest);
    return isSelected(node) ? [ {...node, children: trees} ] : trees;
}

演示：

const demo = { name: "A", children: [
  { name: "B", children: [
    { name: "E", children: [] },
    { name: "F", children: [
      { name: "G", children: [] },
    ] },
  ] },
  { name: "C", children: [] },
  { name: "D", children: [] },
] };
let selection;
function isSelected({name}) {
  return selection.has(name);
}
function selectedForest(node) {
  const trees = node.children.flatMap(selectedForest);
  return isSelected(node) ? [ {...node, children: trees} ] : trees;
}

selection = new Set("ABDEG");
console.log(JSON.stringify(selectedForest(demo), null, 2));
selection = new Set("BDEG");
console.log(JSON.stringify(selectedForest(demo), null, 2));