从 unordered_map 内的向量中删除元素

Question

在我的课堂上，我有一个无序的向量图，如下所示：

std::unordered_map<State, std::vector<std::shared_ptr<const City>>> citiesByState;

我的班级也有这两种方法：

  void addCity(State state, const std::shared_ptr<const City>& city);
  void removeCity(State state, const std::shared_ptr<const City>& city);

我添加一个城市，像这样：

void Manager::addCity(State state, const std::shared_ptr<const City>& city) {
  auto location = citiesByState.find(state); // Find the state in the map
  if (location == citiesByState.end()) { // If the state isn't in the map
    std::vector<std::shared_ptr<const City>> cities; // Create a vector
    cities.push_back(city); // Add the city
    citiesByState[state] = cities; // Add the state and city vector to my map
  } else {
    auto vector = location->second; // Get the city vector. If the city isn't there already, add it.
    if (std::find(vector.begin(), vector.end(), city) == vector.end()) {
      vector.push_back(city);
    }
  }
}

现在这是我删除城市的代码：

void Manager::removeCity(State state, const std::shared_ptr<const City>& city) {
  auto location = citiesByState.find(state);
  if (location != citiesByState.end()) {
    auto vector = location->second;
    if (vector.size() > 0) {
      std::cout << "Vector isn't empty." << std::endl;
    }
    vector.clear(); // Just empty it out for now.
  }
}

然后我像这样运行它：

  City city = ... // get city
  manager->addCity(State::NewYork, city);
  manager->removeCity(State::NewYork, city);

我可以反复调用manager->removeCity(State::NewYork, city)，每次都看到向量不为空。我好像无法从 Vector 中删除。

我做错了什么？

Answer 1

TL;DR

您正在从向量的副本中删除元素，而不是从存在于 std::unordered_map 中找到的 location 中的 std::vector 中。

长篇大论

当您在Manager::removeCity 中调用auto vector = location->second; 时，您将在if 语句的范围内复制该向量。因此，您的更改不会反映在您定位的容器中。只有您的副本会受到影响，并且在 if 语句的末尾也超出了范围，因此如果您找到 location 发生的所有事情都不会保存在 std::unordered_map 容器的状态中。

您可以通过直接调用location->second.clear() 来解决此问题，或者，如果您真的想给它另一个名称，请使用参考，例如auto& vec = location->second; vec.clear();。请注意，这也适用于 Manager::addCity 方法。

附：为了避免混淆，我会避免使用与容器或 STL 中完善的类相同的变量名。