-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTraffic_light.cpp
More file actions
103 lines (86 loc) · 1.8 KB
/
Traffic_light.cpp
File metadata and controls
103 lines (86 loc) · 1.8 KB
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#include<iostream>
#include<vector>
// "Traffic light" task, http://acmp.ru/index.asp?main=task&id_task=124
class Graph
{
public:
typedef size_t Vertex;
Graph(size_t vertex_count, bool is_directed)
{
vertex_count_ = vertex_count;
is_directed_ = is_directed;
}
size_t getVertexCount() const
{
return vertex_count_;
}
size_t getEdgeCount() const
{
return edge_count_;
}
virtual std::vector<Vertex> getNeighbors(const Vertex& v) const = 0;
virtual void addEdge(const Vertex& start, const Vertex& finish)
{
edge_count_++;
}
virtual size_t getNeighborsCount(const Vertex& v) const = 0;
protected:
bool is_directed_;
size_t vertex_count_,
edge_count_;
};
class GraphAdjList : public Graph
{
public:
GraphAdjList(const size_t vertex_count, bool is_directed) :
Graph(vertex_count, is_directed)
{
adj_list_ = std::vector<std::vector<Vertex>>(vertex_count);
}
void addEdge(const Vertex& start, const Vertex& finish)
{
Graph::addEdge(start, finish);
adj_list_[start].push_back(finish);
if (!is_directed_)
adj_list_[finish].push_back(start);
}
std::vector<Vertex> getNeighbors(const Vertex& v) const override
{
return adj_list_[v];
}
size_t getNeighborsCount(const Vertex& v) const override
{
return adj_list_[v].size();
}
void printGraph() const
{
std::cout << "Print\n";
std::cout << vertex_count_;
for (size_t i = 0; i < vertex_count_; i++)
{
for (auto k : adj_list_[i])
std::cout << k << " ";
std::cout << std::endl;
}
}
private:
std::vector<std::vector<Vertex>> adj_list_;
};
int main()
{
int n, m;
std::cin >> n;
std::cin >> m;
GraphAdjList g(n, false);
for (size_t i = 0; i < m; i++)
{
size_t a, b;
std::cin >> a;
std::cin >> b;
g.addEdge(a - 1, b - 1);
}
for (int i = 0; i < n; i++)
{
std::cout << g.getNeighborsCount(i) << " ";
}
}