C++ vector 边遍历边删除的常见陷阱与 remove_if 正确用法详解

你的这段 C++ 代码存在 **严重逻辑错误和运行时未定义行为（UB）**，主要问题如下： --- ### ❌ 主要错误分析 #### 1. **边遍历边 `erase()` 导致索引错位（最严重！）** ```cpp for(int i = 0; i < n; i++){ if(v[i]%2 == 0){ v.erase(v.begin()+i); // ⚠️ 删除后，后续元素前移，但 i 仍自增 → 跳过下一个元素！ } } ``` - 假设 `v = [2, 4, 1, 3]`，当 `i=0` 删除 `2` 后，`v` 变为 `[4,1,3]`，此时 `i` 变为 `1`，**直接跳过了原位置1的 `4`**（现在在索引0），导致漏删甚至越界访问。 - 更糟的是：循环上限 `n` 是**原始大小**，而 `v.size()` 在删除后不断变小，`i` 很可能超出新 `v.size()` → **越界读取 → 未定义行为（崩溃/乱码）** ✅ 正确做法： - 使用 **反向遍历**（`i` 从 `size()-1` 到 `0`），或 - 使用 **迭代器 + `erase()` 返回值**，或 - 使用 **`std::remove_if` + `erase` 惯用法（推荐）** --- #### 2. **未更新 `n`，导致后续 `sort` 和输出逻辑混乱** 你用了 `n` 作为循环上限，但删除后 `v.size()` 已变小，而 `n` 仍是初始值。虽然 `sort(v.begin(), v.end())` 本身安全（它用 `v.size()`），但若你后续还依赖 `n` 就会出错。 --- #### 3. **缺少换行 / 输出格式不规范** `cout << i << " ";` 末尾多一个空格（虽不报错，但通常不推荐）。建议用 `join` 风格或控制末尾空格。 --- ### ✅ 正确且现代的写法（推荐） ```cpp #include <iostream> #include <vector> #include <algorithm> using namespace std; int main() { int n; cin >> n; vector<int> v(n); for (int i = 0; i < n; ++i) { cin >> v[i]; } // ✅ 安全移除所有偶数：remove_if + erase 惯用法（稳定、高效、无越界） v.erase( remove_if(v.begin(), v.end(), [](int x) { return x % 2 == 0; }), v.end() ); // ✅ 排序剩余的奇数 sort(v.begin(), v.end()); // ✅ 输出（避免末尾空格） for (size_t i = 0; i < v.size(); ++i) { if (i > 0) cout << " "; cout << v[i]; } cout << endl; return 0; } ``` --- ### 🔍 补充说明 | 方法 | 是否安全 | 时间复杂度 | 备注 | |------|----------|------------|------| | `remove_if + erase` | ✅ 是 | O(n) | 标准库惯用法，**强烈推荐** | | 反向 for 循环 | ✅ 是 | O(n²) | `erase` 每次移动后续元素，效率低 | | 正向 + 手动调整 `i`（如 `i--`） | ⚠️ 易错 | O(n²) | 容易写错，不推荐 | --- ### 🧪 测试样例验证 **输入：** ``` 5 2 1 4 3 6 ``` **期望输出：** ``` 1 3 ``` ✅ 上述修正代码输出正确；原代码很可能输出 `1 3 6` 或崩溃。 --- ### 💡 开发建议（额外加分项） - 使用 `size_t` 或 `vector<int>::size_type` 替代 `int` 遍历容器（避免有符号/无符号警告）； - 开启编译器警告（如 `-Wall -Wextra -fsanitize=address,undefined`）可提前捕获此类错误； - 学习 STL 算法组合（如 `remove_if`, `partition`, `stable_partition`）比手写循环更安全高效。 --- ######[AI写代码神器 | 1057点数解答 | 2026-03-22 09:23:09]