1 files changed, 28 insertions, 0 deletions

@@ -0,0 +1,28 @@
+#ifndef NETWORK_SET_H_3036B8E3
+#define NETWORK_SET_H_3036B8E3
+
+#include <map>
+#include <optional>
+#include <set>
+#include <utility>
+#include <vector>
+
+class NetworkSet {
+ public:
+  void Clear();
+
+  int AddLink(int id1, int id2);
+
+  bool IsItemInNetwork(int id) const;
+
+  int GetNetworkWithItem(int id) const;
+
+  const std::set<std::pair<int, int>>& GetNetworkGraph(int id) const;
+
+ private:
+
+  std::vector<std::set<std::pair<int, int>>> networks_;
+  std::map<int, int> network_by_item_;
+};
+
+#endif /* end of include guard: NETWORK_SET_H_3036B8E3 */