std::map的[]方法重写

需求

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

阅读更多
文章标签: std-map
个人分类: C++ programming
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭