Question

我有std::vector<T>和std::map<U, std::size_t>，其中地图作为索引矢量的不同方式，即地图的键值对中的值是矢量的索引元件。我可以保证不仅钥匙，而且地图的价值也是唯一的。

现在，当我向矢量应用std :: sort时，这显然会改变可以找到元素的索引，并且我想相应地更新我的地图。

我找到this答案，用于将排列应用于第二个向量。我想要做的是将这个排列应用于地图的值。

以下代码确实有效，但当然非常低效

template <typename T, typename U>
std::vector<T> apply_permutation(const std::vector<T>& vec,
                                 std::map<U, std::size_t>& m,
                                 const std::vector<std::size_t>& p)
{
    std::vector<T> sorted_vec(vec.size());
    std::transform(p.begin(), p.end(), sorted_vec.begin(),
                   [&](std::size_t i){ return vec[i]; });
    for (std::size_t i(0); i != p.size(); ++i)
        for (auto it = m.begin(); it != m.end(); ++it)
            if (it->second == p[i])
                it->second = i;

    return sorted_vec;
}

我正在寻找一种更有效的方法来保持矢量和地图同步。

Answer 1

让我们说从这开始：

vector<char> v1{'b', 'c', 'a'};
map<char, size_t> m{{'b', 0}, {'c', 1}, {'a', 2}};

如果我们将地图从begin()迭代到end()，我们会得到：

  {'a', 2}
  {'b', 0}
  {'c', 1}

因为地图会对键上的元素进行排序。

现在，如果您对矢量进行排序，它将变为

vector<char> v2{'a', 'b', 'c'};

此时，您可以迭代地图并更改索引

size_t index = 0;
for (auto& elem : m) {
  assert(elem.first == v2[index]);
  elem.second = index++;
}

如何将排列应用于std :: map值

1 个答案: