溫馨提示×

如何在std::remove后處理被移除的元素

c++
小樊
86
2024-08-18 01:35:39
欄目: 編程語言

std::remove函數(shù)中,并不會真正移除元素,而是將要移除的元素移到容器的末尾,并返回一個指向新的邏輯結(jié)尾的迭代器。如果要處理被移除的元素,可以使用std::remove_if結(jié)合std::copy_if或者自定義的處理邏輯來實(shí)現(xiàn)。

以下是一種處理被移除的元素的方法:

#include <iostream>
#include <algorithm>
#include <vector>

int main() {
    std::vector<int> vec = {1, 2, 3, 4, 5};
    int valueToRemove = 3;

    auto it = std::remove(vec.begin(), vec.end(), valueToRemove);

    // 使用std::copy_if將被移除的元素復(fù)制到一個新的容器
    std::vector<int> removedElements;
    std::copy_if(vec.begin(), it, std::back_inserter(removedElements), [=](int val) {
        return val == valueToRemove;
    });

    // 打印原始容器
    for (int i : vec) {
        std::cout << i << " ";
    }
    std::cout << std::endl;

    // 打印被移除的元素
    for (int i : removedElements) {
        std::cout << i << " ";
    }
    std::cout << std::endl;

    return 0;
}

在上面的代碼中,我們先使用std::remove將要移除的元素移到容器的末尾,然后使用std::copy_if將被移除的元素復(fù)制到一個新的容器中。最后,打印原始容器和被移除的元素。

0