删除所有具有零值的地图键

时间:2018-02-08 15:46:05

标签: c++ dictionary erase

这张地图的钥匙属于{0,1,2,3}。

我需要擦除所有值等于0的键。

这段代码是一个好习惯吗?

map<int, int> nums = {{0, 1}, {1, 3}, {2, 0}, {3, 1}};

for(int i = 0; i < 4; i++)
    if (nums.count(i) > 0 && nums[i] == 0)
        nums.erase(i);

它似乎可以工作但迭代地图并在同一循环中擦除键让我感到不舒服。

如果这段代码不是很好的方式,那么在地图中删除零值的所有键的最佳方法是什么?

1 个答案:

答案 0 :(得分:6)

这是一个非常接近您的任务http://en.cppreference.com/w/cpp/container/map/erase的好例子 我为你更新了它。

#include <map>
#include <iostream>

int main()
{
    std::map<int, int> c = {{1, 1}, {2, 0}, {3, 3},
                                    {4, 0}, {5, 5}, {6, 0}};
    // erase all key-value pairs with zero values from c
    for(auto it = c.begin(); it != c.end(); )
        if(it->second == 0)
            it = c.erase(it);
        else
            ++it;
    for(auto& p : c)
        std::cout << p.second << ' ';
}

输出:

1 3 5

我建议您更频繁地访问http://en.cppreference.com