m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/graph.h6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/graph.h b/src/graph.h
index b4b300b..7d0ae08 100644
--- a/src/graph.h
+++ b/src/graph.h
@@ -19,6 +19,7 @@ public:
add_vertex(from);
add_vertex(to);
graph_[from].push_back(to);
+ has_out_edges_.insert(from);
}
const std::set<int> & get_vertices() const {
@@ -28,9 +29,14 @@ public:
const std::vector<int> & get_neighbors(int vertex) const {
return graph_.find(vertex)->second;
}
+
+ bool has_out_edges(int vertex) const {
+ return has_out_edges_.count(vertex) > 0;
+ }
private:
std::set<int> vertices_;
std::map<int, std::vector<int>> graph_;
+ std::set<int> has_out_edges_;
};
#endif