STL中的容器可以分为两类:一类是按照数组方式连续存储元素的容器,如vector
、deque
,另一类是按照不连续的节点存储元素的容器,如set
、map
、list
等。
那么这两类容器删除元素的时候需要注意一个问题,使用iterator
指定删除某个元素的时候,要注意返回的iterator
的值。下面分别举set
、vector
的例子来说明。
set删除元素
假如要删除set中满足某个条件的元素,这时候需要遍历,看代码:
set<int> s;
s.insert(1);
s.insert(4);
s.insert(2);
s.insert(3);
for (set<int>::iterator it = s.begin(); it != s.end(); it ++)
cout << *it << endl;
for (set<int>::iterator it = s.begin(); it != s.end(); )
{
if (*it % 2 != 0) // 删除奇数
s.erase(it++);
else
it ++;
}
大家注意到s.erase(it++)
这一句,it++
返回是+1之前的it
的值,那么此时删除的正是满足条件的元素,而it
本身又进行了+1,指向了下一个元素,那你可能会问为什么不直接删除之后再+1呢,如下代码:
s.erase(it);
it++;
这样为什么不行呢?因为在erase
之前,it
其实指向的还是原来的元素,只不过它已经不存在set
中,那么此时it++
,得到的值不是set
中的那一个元素,而是不确定的,对这个指针进行访问,会得到意想不到的结果,甚至导致程序崩溃。
vector删除元素
vector
内部存储元素是采用数组方式存储的,即是连接的,删除元素可以按照以下方法:
vector<int> ve;
ve.push_back(1);
ve.push_back(3);
ve.push_back(2);
ve.push_back(4);
for (vector<int>::iterator it = ve.begin(); it != ve.end();)
{
if (*it % 2 != 0)
ve.erase(it);
else
it ++;
}
vector
跟set
不一样的地方在于,调用erase
之后,it
自动指向下一个元素,所以在遍历删除的时候需要注意,erase
之后不需要再进行it++
,否则会跳过了一个元素。
网友评论