JavaScript实现：从数组A中仅移除数组B各元素的一个实例

我来帮你梳理下代码里的问题，看看为什么达不到预期的1,3,3,4,5输出效果～

你的代码里的核心问题 #

• 未定义变量报错：你代码里多次用到的a是未声明的变量，应该替换成函数参数里的arrayA，不然函数执行时会直接抛出错误，根本没法运行。
• 参数顺序易混淆：你定义的函数是remove(arrayB, arrayA)，但逻辑上是要处理arrayA、移除arrayB中的元素，参数顺序反过来会增加理解成本，建议调整为remove(arrayA, arrayB)。
• 逻辑分支冗余+遗漏调用：原代码的分支判断过于复杂，而且最后没有调用函数，导致不会执行输出逻辑。

修正后的简化版代码 #

先基于你的思路调整，修复问题后可以得到正确结果：

let arrayA = [1,1,2,3,3,3,4,4,5]
let arrayB = [1,2,3,4]

function remove(arrayA, arrayB){ 
    let newarray = [];
    // 用数组记录已经移除过一次的元素
    let removedItems = [];
    arrayA.forEach(function(n){
        // 如果元素不在B里，直接保留
        if(arrayB.indexOf(n) === -1){
            newarray.push(n);
        } else {
            // 如果该元素还没被移除过一次，就跳过（相当于移除）
            if(!removedItems.includes(n)){
                removedItems.push(n);
            } else {
                // 已经移除过一次，后续实例全部保留
                newarray.push(n);
            }
        }
    })
    console.log(newarray.join(',')); // 输出预期的 1,3,3,4,5
}

// 别忘了调用函数！
remove(arrayA, arrayB);

更高效的优化写法 #

如果数组规模较大，上面的indexOf和includes性能会比较差，因为每次查找都是O(n)时间复杂度。可以用Map来记录需要移除的元素次数，把查找操作降到O(1)：

let arrayA = [1,1,2,3,3,3,4,4,5]
let arrayB = [1,2,3,4]

function remove(arrayA, arrayB) {
    // 记录每个元素需要移除的次数（每个B中的元素仅移除1次）
    const removeMap = new Map();
    arrayB.forEach(item => removeMap.set(item, 1));

    const result = [];
    arrayA.forEach(item => {
        const needRemove = removeMap.get(item);
        if (needRemove && needRemove > 0) {
            // 还需要移除一次，跳过该元素并更新计数
            removeMap.set(item, needRemove - 1);
        } else {
            // 无需移除，加入结果数组
            result.push(item);
        }
    });

    console.log(result.join(',')); // 输出 1,3,3,4,5
}

remove(arrayA, arrayB);

内容的提问来源于stack exchange，提问作者Noémie Carette