【C++】如何修改set的值

问题：尝试通过begin方法得到的迭代器去修改值，发现会报错。

set<string> st{"hello", "world", "good"};
set<string>::iterator it = st.begin();
*it = "test";

原因：我们可以在源码里找到问题根源

原来迭代器的定义是一个const_iterator，搞了半天，其实并不是set底层限制了禁止修改，而是通过迭代器限制的。所以要想修改set当中的元素，我们只需要绕开迭代器的这个限制即可。

解决办法：

（1）用const_cast去掉const属性

set<string> st{"hello", "world", "good"};
set<string>::iterator it = st.begin();
const_cast<string&>(*it) = "test";    //it是迭代器，可以看成指针

for (auto it = st.begin(); it != st.end(); it++) {
    cout << *it << endl;
}

但是我们需要注意一点，我们这样强行修改其实是没有经过set原本的机制的。也就是说我们虽然改了元素的值，但是它在红黑树中的位置其实是没有变的。这样的结果就是会导致元素失去有序性，比如上面的结果输出的顺序是："test","hello","world"，按道理应该是按照字典顺序排序的。

（2）可以通过insert_iterator函数把迭代器强转为可插入形式

string & ToLower(string & st)
{
    transform(st.begin() , st.end() , st.begin() , tolower);
    return st;
}

set<string> wordset;
transform(words.begin() , words.end() , 
            insert_iterator< set<string> > (wordset , wordset.begin() ) , ToLower);

//这里tolower和ToLower都是STL的函数符

【C++】如何修改set的值

猜你喜欢