Merge branch 'subway'

1 files changed, 30 insertions, 0 deletions

@@ -0,0 +1,30 @@
+#include "network_set.h"
+
+void NetworkSet::Clear() {
+  network_by_item_.clear();
+}
+
+void NetworkSet::AddLink(int id1, int id2) {
+  if (id2 > id1) {
+    // Make sure id1 < id2
+    std::swap(id1, id2);
+  }
+
+  if (!network_by_item_.count(id1)) {
+    network_by_item_[id1] = {};
+  }
+  if (!network_by_item_.count(id2)) {
+    network_by_item_[id2] = {};
+  }
+
+  network_by_item_[id1].insert({id1, id2});
+  network_by_item_[id2].insert({id1, id2});
+}
+
+bool NetworkSet::IsItemInNetwork(int id) const {
+  return network_by_item_.count(id);
+}
+
+const std::set<std::pair<int, int>>& NetworkSet::GetNetworkGraph(int id) const {
+  return network_by_item_.at(id);
+}