（无序）映射键和值修改的最佳实践

Question

给出map<long long, vector<long long>> 形式的映射。必须将所有键和值取模某个整数N。一些键可以合并，相应的值必须相应地加入。例如，映射{{1,{2,6,4}}, {5,{8,4,9}}, {10,{5,1,7}}} 应该等于{{1,{2,1,4}}, {0,{0,1,2,3,4}}} 减少模5。

我的方法是使用新地图，但我认为应该有更好的方法。

添加代码

vector<long long> tmp;
//integer N, for example N = 5
int N = 5;
unordered_map<long long, vector<long long>> map;
//temporary map
unordered_map<long long, vector<long long>> map_tmp;
     for (auto & x : map)
        {
            tmp.clear();
            for (auto & y : x.second) tmp.push_back(y % N);
            ind = x.first % N;
            map_tmp[ind].insert(map_tmp[ind].end(), tmp.begin(), tmp.end());
            sort(map_tmp[ind].begin(), map_tmp[ind].end());
            map_tmp[ind].erase(unique(map_tmp[ind].begin(), map_tmp[ind].end()), map_tmp[ind].end());
        }
        map = map_tmp;

Answer 1

由于 map 中的值显然是唯一的，并且在应用模运算后值包含唯一项，因此您应该使用不同的数据结构。例如：

using Map = std::unordered_map<int, std::set<int>>;

std::set 将处理给定键的项目的唯一性和顺序。

现在整个技巧是检查std::unordered_map 和std::set 的API 以及如何将项目插入那里。见：

• std::unordered_map::insert
• std::set::insert

注意返回值：std::pair<iterator,bool>，它为您提供了在地图/集合中插入或令人兴奋的项目的迭代器。

知道编写能够满足您要求的代码非常简单：

using Map = std::unordered_map<int, std::set<int>>;

Map moduloMap(const Map& in, int mod)
{
    Map out;
    for (const auto& [k, s] : in) {
        if (s.empty())
            continue;
        auto& destSet = out.insert({ k % mod, {} }).first->second;
        for (auto x : s) {
            destSet.insert(x % mod);
        }
    }
    return out;
}

Live demo with tests

Answer 2

有时for 循环可能是做某事的最简单、最清晰的方法。

map<long long, vector<long long>> result;
for (const auto& [key, vec] : input) {
    process (result[key%5], vec);
    }

而process 通过（非常量）引用获取vector，并附加第二个（常量）参数的缩减值。

---

更新

看了你贴的代码，我有几个建议：

1. 请改用set。您正在花费多个步骤来附加新值，将整个内容排序在一起，然后删除重复项。只需使用 set 即可自动维护每个值的单个副本。
2. 在循环中使用结构化绑定。除了x.second 和x.first，您只需将它们命名为key 和vec，就像我之前的帖子中一样。
3. 假设您仍然需要tmp，请在您现在调用.clear() 的位置声明它，而不是在代码顶部声明它。您不需要每次通过循环都清除它；每次循环自然都会为空。