src/graph.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

#ifndef GRAPH_H
#define GRAPH_H

#include <map>
#include <vector>

class Graph {
public:
    Graph() : graph_() {}

    void add_vertex(int vertex) {
        graph_[vertex] = std::vector<int>();
    }

    void add_edge(int from, int to) {
        graph_[from].push_back(to);
    }

    std::set<int> get_vertices() {
        std::set<int> vertices;

        for (auto vertex : graph_) {
            vertices.insert(vertex.first);
        }

        return vertices;
    }


    std::vector<int> const& get_neighbors(int vertex) {
        return graph_[vertex];
    }
private:
    std::map<int, std::vector<int>> graph_;
};

#endif