1. 程式人生 > >STL演算法之remove要注意的地方。以及迭代時如何正確的刪除迭代器

STL演算法之remove要注意的地方。以及迭代時如何正確的刪除迭代器

    std::array<int, 6> intArray = {1, 2, 3, 4, 5, 6};
    std::vector<int> intVector;

    for(auto& i : intArray)
        intVector.push_back(i);

    cout << "intVector Size=" << intVector.size() << endl;

    intVector[0] = intVector[2] = intVector[4] = 99;
//    std::remove(intVector.begin(), intVector.end(), 99);  // 錯誤的用法!!!
    cout << "intVector Size=" << intVector.size() << endl;

    // 移除vector中所有值為99的元素
    intVector.erase(std::remove(intVector.begin(), intVector.end(), 99), intVector.end());
    cout << "intVector Size=" << intVector.size() << endl;

    // 迭代的時候刪除迭代器,有可能導致迭代器失效。以下是正確的刪除vector的迭代器的做法
    // 因為呼叫vector的erase方法後,返回的是下一個迭代器,需要手工更新迭代器
    for(std::vector<int>::const_iterator cit = intVector.begin(); cit != intVector.end();)
    {
        cout << *cit << endl;
        cit = intVector.erase(cit);
    }