C++从vector中删除非质数时函数仅能移除0和1问题求解

你的代码存在以下几处核心逻辑错误，是导致无法正确过滤非质数的原因：

• 删除小于等于1的元素时遍历逻辑错误：按索引遍历vector过程中调用erase会使vector长度缩减，后续元素前移，此时w直接递增会跳过下一个待检查元素，极端场景下会漏删部分小于等于1的元素。
• 遍历过程修改vector导致迭代器失效：你使用范围for循环遍历foo，但循环内调用erase删除元素会破坏原迭代器，触发未定义行为。
• 除数计数变量未重置：limit_counter仅在函数初始化时赋值为1，处理完第一个元素后就会增长到接近p_limit的值，后续元素判断时不会重新从最小除数开始校验，这是你只能删除0和1的核心原因。
• 质数判断规则错误：质数的定义是大于1，且仅能被1和自身整除的自然数。你从1开始做模运算，所有数模1结果都是0，会误删所有质数；同时没有排除自身作为除数的判断规则，逻辑完全不符合质数判定要求。
• 头文件引用错误：标准库算法头文件为<algorithm>，数学库头文件为<cmath>，你当前的引用写法不符合C++标准规范。

由于你传入的vector是升序排列的连续整数，直接使用埃拉托斯特尼筛法实现效率最高，逻辑也更清晰，修复后的代码如下：

#include <vector>
#include <algorithm>
#include <cmath>

std::vector<int> prime_list(std::vector<int> foo) {
    // 用erase-remove idiom安全删除所有小于等于1的元素
    foo.erase(std::remove_if(foo.begin(), foo.end(), [](int num) {
        return num <= 1;
    }), foo.end());
    if (foo.empty()) {
        return foo;
    }

    int max_val = foo.back();
    std::vector<bool> is_prime(max_val + 1, true);
    is_prime[0] = is_prime[1] = false;
    // 埃氏筛标记非质数
    for (int i = 2; i <= static_cast<int>(std::sqrt(max_val)); ++i) {
        if (is_prime[i]) {
            for (int j = i * i; j <= max_val; j += i) {
                is_prime[j] = false;
            }
        }
    }

    // 过滤所有非质数
    foo.erase(std::remove_if(foo.begin(), foo.end(), [&](int num) {
        return !is_prime[num];
    }), foo.end());
    return foo;
}

如果你坚持要沿用逐个判定的原有逻辑，需要给每个待判定的元素重置除数计数器，从2开始校验，同时避免边遍历边修改原vector：可以先收集所有质数到新vector再返回，避免迭代器失效问题。

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