js树形

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

  addChild(childNode) {
    this.children.push(childNode);
  }
}

// 创建根节点
const root = new TreeNode('Root');

// 添加子节点
const child1 = new TreeNode('Child 1');
const child2 = new TreeNode('Child 2');
root.addChild(child1);
root.addChild(child2);

// 添加孙节点
const grandchild1 = new TreeNode('Grandchild 1');
child1.addChild(grandchild1);

console.log(root);

function traverse(node) {
  if (!node) return;
  console.log(node.value); // 访问当前节点
  node.children.forEach(child => traverse(child)); // 递归访问子节点
}

traverse(root);

function findNode(node, value) {
  if (!node) return null;
  if (node.value === value) return node;
  for (let child of node.children) {
    const result = findNode(child, value);
    if (result) return result;
  }
  return null;
}

const foundNode = findNode(root, 'Grandchild 1');
console.log(foundNode);

function removeNode(node, value) {
  if (!node) return;
  node.children = node.children.filter(child => {
    if (child.value === value) {
      return false; // 删除匹配的节点
    } else {
      removeNode(child, value); // 递归删除子节点中的匹配节点
      return true;
    }
  });
}

removeNode(root, 'Child 1');
console.log(root);