95 lines
1.8 KiB
C++
95 lines
1.8 KiB
C++
#ifndef GRAPH_H_
|
|
#define GRAPH_H_
|
|
|
|
#include <map>
|
|
#include <set>
|
|
#include <iostream>
|
|
|
|
namespace graph {
|
|
|
|
template<typename T>
|
|
class Graph {
|
|
public:
|
|
~Graph();
|
|
|
|
// Add edge e(u,v)
|
|
virtual void insert(const T& u, const T& v);
|
|
|
|
// Return true if e(u,v) exists
|
|
virtual bool contains(const T& u, const T& v);
|
|
|
|
// Return true if vertex u exists
|
|
virtual bool contains(const T& u);
|
|
|
|
// Remove edge e(u,v)
|
|
virtual void remove(const T& u, const T& v);
|
|
|
|
// Return num. of vertices
|
|
virtual std::uint16_t V();
|
|
|
|
// Return num. of edges
|
|
virtual std::uint16_t E();
|
|
|
|
// Output graph
|
|
virtual void output();
|
|
|
|
// Adjacency matrix representation
|
|
std::map<T, std::set<T>> adjMatrix;
|
|
};
|
|
|
|
template<typename T>
|
|
Graph<T>::~Graph() {
|
|
adjMatrix.clear();
|
|
}
|
|
|
|
template<typename T>
|
|
inline void Graph<T>::insert(const T& u, const T& v) {
|
|
Graph<T>::adjMatrix[u].insert(v);
|
|
Graph<T>::adjMatrix[v];
|
|
|
|
}
|
|
|
|
template<typename T>
|
|
bool Graph<T>::contains(const T& u, const T& v) {
|
|
return adjMatrix[u].contains(v);
|
|
}
|
|
|
|
template<typename T>
|
|
bool Graph<T>::contains(const T& u) {
|
|
return adjMatrix.contains(u);
|
|
}
|
|
|
|
template<typename T>
|
|
void Graph<T>::remove(const T& u, const T& v) {
|
|
adjMatrix[u].erase(v);
|
|
}
|
|
|
|
template<typename T>
|
|
std::uint16_t Graph<T>::V() {
|
|
return static_cast<std::uint16_t>(adjMatrix.size());
|
|
}
|
|
|
|
template<typename T>
|
|
std::uint16_t Graph<T>::E() {
|
|
std::uint16_t edges = 0;
|
|
for (const auto& v : adjMatrix) {
|
|
edges += static_cast<std::uint16_t>(v.second.size());
|
|
}
|
|
return static_cast<std::uint16_t>(edges);
|
|
}
|
|
|
|
template<typename T>
|
|
void Graph<T>::output() {
|
|
for (const auto& v : adjMatrix) {
|
|
for (const auto& u : v.second) {
|
|
std::cout << v.first << "->" << u << '|';
|
|
}
|
|
std::cout << '\n';
|
|
}
|
|
std::cout << '\n';
|
|
}
|
|
|
|
} // namespace graph
|
|
|
|
|
|
#endif |