ฉันต้องผ่านชุดและลบองค์ประกอบที่ตรงตามเกณฑ์ที่กำหนดไว้ล่วงหน้า
นี่คือรหัสทดสอบที่ฉันเขียน:
#include <set>
#include <algorithm>
void printElement(int value) {
std::cout << value << " ";
}
int main() {
int initNum[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
std::set<int> numbers(initNum, initNum + 10);
// print '0 1 2 3 4 5 6 7 8 9'
std::for_each(numbers.begin(), numbers.end(), printElement);
std::set<int>::iterator it = numbers.begin();
// iterate through the set and erase all even numbers
for (; it != numbers.end(); ++it) {
int n = *it;
if (n % 2 == 0) {
// wouldn't invalidate the iterator?
numbers.erase(it);
}
}
// print '1 3 5 7 9'
std::for_each(numbers.begin(), numbers.end(), printElement);
return 0;
}
ตอนแรกฉันคิดว่าการลบองค์ประกอบออกจากชุดในขณะที่วนซ้ำมันจะทำให้ตัววนซ้ำใช้ไม่ได้และการเพิ่มที่ห่วงสำหรับ for จะมีพฤติกรรมที่ไม่ได้กำหนดไว้ แม้ว่าฉันจะรันโค้ดทดสอบนี้และทุกอย่างเป็นไปด้วยดีและฉันไม่สามารถอธิบายได้ว่าทำไม
คำถามของฉัน: นี่เป็นพฤติกรรมที่กำหนดไว้สำหรับชุด std หรือการใช้งานนี้เฉพาะ ฉันใช้ gcc 4.3.3 บน Ubuntu 10.04 (รุ่น 32 บิต) โดยวิธีการ
ขอบคุณ!
โซลูชันที่เสนอ:
นี่เป็นวิธีที่ถูกต้องในการทำซ้ำและลบองค์ประกอบออกจากชุดหรือไม่
while(it != numbers.end()) {
int n = *it;
if (n % 2 == 0) {
// post-increment operator returns a copy, then increment
numbers.erase(it++);
} else {
// pre-increment operator increments, then return
++it;
}
}
แก้ไข: โซลูชันที่ได้รับการแนะนำ
ฉันมาหาทางแก้ปัญหาที่ดูสง่างามกว่าสำหรับฉันแม้ว่ามันจะทำแบบเดียวกันก็ตาม
while(it != numbers.end()) {
// copy the current iterator then increment it
std::set<int>::iterator current = it++;
int n = *current;
if (n % 2 == 0) {
// don't invalidate iterator it, because it is already
// pointing to the next element
numbers.erase(current);
}
}
หากมีเงื่อนไขการทดสอบหลายอย่างในขณะนั้นแต่ละเงื่อนไขจะต้องเพิ่มตัววนซ้ำ ฉันชอบรหัสนี้ดีกว่าเนื่องจากตัววนซ้ำเพิ่มขึ้นในที่เดียวเท่านั้นทำให้รหัสผิดพลาดน้อยลงและอ่านง่ายขึ้น
++it
ควรมีประสิทธิภาพมากกว่าit++
เพราะไม่จำเป็นต้องใช้ตัวคัดลอกชั่วคราวที่มองไม่เห็นของ iterator เวอร์ชันของ Kornel นั้นจะทำให้แน่ใจได้ว่าองค์ประกอบที่ไม่ผ่านการกรองจะทำซ้ำได้อย่างมีประสิทธิภาพสูงสุด