需求
要求对std::map的[]操作,新插入元素时设定为指定初始值。
实现
std::map<int, int> test_dict;
uint32_t operator[](const int key)
{
// _where指向_where->first比key大于等于的元素
std::map<int, int>::iterator _where = test_dict.lower_bound(key);
// test_dict为空时,_where为end;key插入位置在中间时,key < (*_where).first)
if (_where == test_dict.end() || test_dict.key_comp()(key, (*_where).first))
_where = test_dict.insert(_where, std::pair<int, int>(key, 10000))
return (*_where).second;
}
在中间的情形,比如:
1、<1,5,8>插入6,那么_where指向8,此时test_dict.key_comp()(key, (*_where).first)条件满足,因此在节点8前面插入;
2、<1,5,8>插入8,那么_where指向8,此时test_dict.key_comp()(key, (*_where).first)条件不满足(8不小于8),因此不会新插入。事实上,[]也作为取值的操作符,此时就直接返回了已有元素8的对应的second值了。