多维嵌套数组按ID查找替换元素的代码问题求助

需求 #

在多维嵌套数组中根据ID值查找指定元素，将该元素从数组中移除并替换为空数组，同时返回被移除的元素。

给定的嵌套数组结构 #

{
    id: 0,
    parent: 'p1',
    children: [
        {
            id: 1,
            parent: 'p2',
            children: [
                {
                    id: 3,
                    parent: 'p4',
                    children: []
                },
            ]
        },
        {
            id: 2,
            parent: 'p3',
            children: [
                {
                    id: 4,
                    parent: 'p5',
                    children: []
                },
            ]
        }
    ]
}

示例执行与预期结果 #

执行代码：

let childTobeReplaced = childrenCollector(4, treeData.children);

预期数组结果 #

{
    id: 0,
    parent: 'p1',
    children: [
        {
            id: 1,
            parent: 'p2',
            children: [
                {
                    id: 3,
                    parent: 'p4',
                    children: []
                },
            ]
        },
        {
            id: 2,
            parent: 'p3',
            children: []
        }
    ]
}

预期返回值 #

{
    id: 4,
    parent: 'p5',
    children: []
}

当前实现的问题 #

以下是当前的实现代码：

function childrenCollector(sourceId, nestedarray) {
    for (let index = 0; index < nestedarray.length; index++) {
        console.log(nestedarray[index].id)
        if (nestedarray[index].id === sourceId) {
            let childArray = nestedarray[index];
            nestedarray[index] = []
            return childArray;
        }
        if (nestedarray[index].children.length > 0) {
            return childrenCollector(sourceId, nestedarray[index].children);
        }else{

        }
    }
}

当前函数只能遍历到id为1和3的元素，无法找到其余元素——核心问题是：遍历第一个元素（id=1）时，发现它有子节点就直接递归并return，导致循环直接终止，根本没机会遍历后续的兄弟元素（比如id=2的节点）。

修正方案 #

调整递归逻辑，不要在第一次递归时直接return，而是先接收递归结果，若找到元素则返回，否则继续循环遍历下一个元素：

function childrenCollector(sourceId, nestedarray) {
    for (let index = 0; index < nestedarray.length; index++) {
        const current = nestedarray[index];
        // 检查当前元素是否是目标
        if (current.id === sourceId) {
            const removed = current;
            // 将当前位置替换为空数组
            nestedarray[index] = [];
            return removed;
        }
        // 递归遍历子节点，接收返回结果
        const result = childrenCollector(sourceId, current.children);
        // 如果递归找到了目标元素，直接返回
        if (result) {
            return result;
        }
    }
    // 遍历完所有元素都没找到，返回null
    return null;
}

关键修改点 #

1. 递归调用时不再直接return，而是先把结果存到result变量中，只有当result存在（找到目标元素）时才return，否则继续循环下一个兄弟元素。
2. 移除了不必要的else分支，简化逻辑。
3. 增加了遍历完所有元素未找到时的返回值（null），避免隐式返回undefined。

内容的提问来源于stack exchange，提问作者gyan mishra