std::map的[]方法重写

xiaoxiao2021-02-28  105

需求

要求对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值了。

转载请注明原文地址: https://www.6miu.com/read-85773.html

最新回复(0)