diff options
Diffstat (limited to 'src/network_set.h')
| -rw-r--r-- | src/network_set.h | 7 |
1 files changed, 2 insertions, 5 deletions
| diff --git a/src/network_set.h b/src/network_set.h index e0ef043..e6f0c07 100644 --- a/src/network_set.h +++ b/src/network_set.h | |||
| @@ -11,18 +11,15 @@ class NetworkSet { | |||
| 11 | public: | 11 | public: |
| 12 | void Clear(); | 12 | void Clear(); |
| 13 | 13 | ||
| 14 | int AddLink(int id1, int id2); | 14 | void AddLink(int id1, int id2); |
| 15 | 15 | ||
| 16 | bool IsItemInNetwork(int id) const; | 16 | bool IsItemInNetwork(int id) const; |
| 17 | 17 | ||
| 18 | int GetNetworkWithItem(int id) const; | ||
| 19 | |||
| 20 | const std::set<std::pair<int, int>>& GetNetworkGraph(int id) const; | 18 | const std::set<std::pair<int, int>>& GetNetworkGraph(int id) const; |
| 21 | 19 | ||
| 22 | private: | 20 | private: |
| 23 | 21 | ||
| 24 | std::vector<std::set<std::pair<int, int>>> networks_; | 22 | std::map<int, std::set<std::pair<int, int>>> network_by_item_; |
| 25 | std::map<int, int> network_by_item_; | ||
| 26 | }; | 23 | }; |
| 27 | 24 | ||
| 28 | #endif /* end of include guard: NETWORK_SET_H_3036B8E3 */ | 25 | #endif /* end of include guard: NETWORK_SET_H_3036B8E3 */ |
