การลบองค์ประกอบจากเวกเตอร์

ฉันต้องการล้างองค์ประกอบจากเวกเตอร์โดยใช้วิธีการลบ แต่ปัญหาที่นี่คือไม่รับประกันว่าองค์ประกอบจะเกิดขึ้นเพียงครั้งเดียวในเวกเตอร์ อาจมีอยู่หลายครั้งและฉันจำเป็นต้องล้างทั้งหมด รหัสของฉันเป็นดังนี้:

void erase(std::vector<int>& myNumbers_in, int number_in)
{
    std::vector<int>::iterator iter = myNumbers_in.begin();
    std::vector<int>::iterator endIter = myNumbers_in.end();
    for(; iter != endIter; ++iter)
    {
        if(*iter == number_in)
        {
            myNumbers_in.erase(iter);
        }
    }
}

int main(int argc, char* argv[])
{
    std::vector<int> myNmbers;
    for(int i = 0; i < 2; ++i)
    {
        myNmbers.push_back(i);
        myNmbers.push_back(i);
    }

    erase(myNmbers, 1);

    return 0;
}

เห็นได้ชัดว่าโค้ดนี้ขัดข้องเพราะฉันกำลังเปลี่ยนจุดสิ้นสุดของเวกเตอร์ขณะที่ทำซ้ำ วิธีที่ดีที่สุดในการบรรลุเป้าหมายนี้คืออะไร? มีวิธีใดบ้างที่จะทำสิ่งนี้โดยไม่ต้องทำซ้ำผ่านเวกเตอร์หลาย ๆ ครั้งหรือสร้างสำเนาเวกเตอร์อีกหนึ่งชุด

ใช้สำนวนลบ / ลบ :

std::vector<int>& vec = myNumbers; // use shorter name
vec.erase(std::remove(vec.begin(), vec.end(), number_in), vec.end());

สิ่งที่เกิดขึ้นคือการremoveบีบอัดองค์ประกอบที่แตกต่างจากค่าที่จะลบ ( number_in) ในจุดเริ่มต้นของvectorและส่งคืนตัววนซ้ำไปยังองค์ประกอบแรกหลังจากช่วงนั้น จากนั้นeraseลบองค์ประกอบเหล่านี้ (ซึ่งไม่ได้ระบุค่า)

การลบการโทรจะทำให้ตัววนซ้ำไม่ถูกต้องคุณสามารถใช้:

void erase(std::vector<int>& myNumbers_in, int number_in)
{
    std::vector<int>::iterator iter = myNumbers_in.begin();
    while (iter != myNumbers_in.end())
    {
        if (*iter == number_in)
        {
            iter = myNumbers_in.erase(iter);
        }
        else
        {
           ++iter;
        }
    }

}

หรือคุณสามารถใช้std :: remove_ifร่วมกับ functor และ std :: vector :: erase:

struct Eraser
{
    Eraser(int number_in) : number_in(number_in) {}
    int number_in;
    bool operator()(int i) const
    {
        return i == number_in;
    }
};

std::vector<int> myNumbers;
myNumbers.erase(std::remove_if(myNumbers.begin(), myNumbers.end(), Eraser(number_in)), myNumbers.end());

แทนที่จะเขียน functor ของคุณเองในกรณีนี้คุณสามารถใช้std :: remove :

std::vector<int> myNumbers;
myNumbers.erase(std::remove(myNumbers.begin(), myNumbers.end(), number_in), myNumbers.end());

ใน C ++ 11 คุณสามารถใช้ lambda แทน functor ได้:

std::vector<int> myNumbers;
myNumbers.erase(std::remove_if(myNumbers.begin(), myNumbers.end(), [number_in](int number){ return number == number_in; }), myNumbers.end());

ใน C ++ 17 มาตรฐาน :: ทดลอง :: ลบและมาตรฐาน :: ทดลอง :: erase_ifนอกจากนี้ยังมีใน C ++ 20 เหล่านี้เป็น (สุดท้าย) เปลี่ยนชื่อเป็นมาตรฐาน :: ลบและมาตรฐาน :: erase_if :

std::vector<int> myNumbers;
std::erase_if(myNumbers, Eraser(number_in)); // or use lambda

หรือ:

std::vector<int> myNumbers;
std::erase(myNumbers, number_in);

1. คุณสามารถทำซ้ำได้โดยใช้การเข้าถึงดัชนี

2. เพื่อหลีกเลี่ยงความซับซ้อน O (n ^ 2) คุณสามารถใช้ดัชนีสองดัชนี i - ดัชนีการทดสอบปัจจุบัน j - ดัชนีเพื่อจัดเก็บรายการถัดไปและเมื่อสิ้นสุดรอบขนาดใหม่ของเวกเตอร์

รหัส:

void erase(std::vector<int>& v, int num)
{
  size_t j = 0;
  for (size_t i = 0; i < v.size(); ++i) {
    if (v[i] != num) v[j++] = v[i];
  }
  // trim vector to new size
  v.resize(j);
}

ในกรณีเช่นนี้คุณไม่มีตัวทำซ้ำที่ไม่ถูกต้องความซับซ้อนคือ O (n) และโค้ดมีความกระชับมากและคุณไม่จำเป็นต้องเขียนคลาสตัวช่วยบางอย่างแม้ว่าในบางกรณีการใช้คลาสตัวช่วยจะได้รับประโยชน์จากโค้ดที่ยืดหยุ่นกว่า

รหัสนี้ไม่ใช้eraseวิธีการ แต่ช่วยแก้งานของคุณ

การใช้ pure stl คุณสามารถทำได้ด้วยวิธีต่อไปนี้ (คล้ายกับคำตอบของ Motti):

#include <algorithm>

void erase(std::vector<int>& v, int num) {
    vector<int>::iterator it = remove(v.begin(), v.end(), num);
    v.erase(it, v.end());
}

ขึ้นอยู่กับว่าทำไมคุณถึงทำสิ่งนี้การใช้std :: setอาจเป็นแนวคิดที่ดีกว่า std :: vector

ช่วยให้แต่ละองค์ประกอบเกิดขึ้นเพียงครั้งเดียว หากคุณเพิ่มหลายครั้งจะมีเพียงอินสแตนซ์เดียวที่จะลบได้ สิ่งนี้จะทำให้การลบข้อมูลเป็นเรื่องเล็กน้อย การดำเนินการลบจะมีความซับซ้อนของเวลาต่ำกว่าเวกเตอร์ด้วยอย่างไรก็ตามการเพิ่มองค์ประกอบจะช้ากว่าในชุดดังนั้นจึงอาจไม่ได้เปรียบมากนัก

แน่นอนว่าวิธีนี้จะใช้ไม่ได้หากคุณสนใจว่าจะมีการเพิ่มองค์ประกอบลงในเวกเตอร์ของคุณกี่ครั้งหรือลำดับการเพิ่มองค์ประกอบ

ในการลบองค์ประกอบที่ 1 คุณสามารถใช้:

vector<int> mV{ 1, 2, 3, 4, 5 }; 
vector<int>::iterator it; 

it = mV.begin(); 
mV.erase(it);