Làm cách nào để lặp qua bộ STL và xóa các phần tử có chọn lọc?

Question

Mã sau đây không hoạt động chính xác. Làm thế nào nó nên được thực hiện một cách chính xác?

for (std::set<Color>::iterator i = myColorContainer.begin(); 
      i!=myColorContainer.end(); 
      ++i) 
{ 
    if (*i == Yellow) 
    { 
     DoSomeProccessing(*i); 
     myColorContainer.erase(i); 
    } 
} 

Answer 1

Bạn không cần một vòng lặp như bạn đang đối phó với một bộ.

std::set<Color>::iterator it = myColorContainer.find(Yellow); 
if (it != it.myColorContainer.end()){ 
    DoSomeProcessing(*it); 
    myColorContainer.erase(it); 
} 

Answer 2

Hãy thử:

for(std::set<Color>::iterator it = myColorContainer.begin(); 
    it != myColorContainer.end();) { // note missing it++ 
    if((*it) == Yellow) { 
     DoSomeProcessing(*it); 
     myColorContainer.erase(it++); // post increment (original sent to erase) 
    } 
    else { 
     ++it; // more efficient than it++; 
    } 
} 

Answer 3

for (std::set<Color>::iterator i = myColorContainer.begin(); 
      i!=myColorContainer.end(); /* No i++ */) 
{ 
    if (*i == Yellow) 
    { 
     DoSomeProccessing(*i); 
     std::set<Color>::iterator tmp = i; 
     ++i; 
     myColorContainer.erase(tmp); 
    } 
    else { 
     ++i; 
    } 
} 

Khi bạn đi đến thông điệp tiếp theo với ++i nó được đảm bảo để có giá trị - tài sản của std::set đó lặp trên các yếu tố chèn không bao giờ chấm dứt hiệu lực trừ khi các yếu tố được lấy ra.

Vì vậy, bây giờ bạn có thể xóa an toàn mục nhập trước đó.