a01sa01to's competitive programming library.
#include "library/data-structure/digraph.hpp"
有向グラフを扱う。
digraph(n)
: $n$ 頂点の有向グラフを構築する。 $O(n)$add_edge(from, to)
: 頂点 from
から頂点 to
への辺を追加する。 $O(1)$get_edge(i)
: 辺 i
の情報を取得する。 $O(1)$get_adj(i)
: 頂点 i
からの隣接リスト (頂点番号, 辺番号) を取得する。 $O(1)$?get_adjrev(i)
: 頂点 i
への隣接リスト (頂点番号, 辺番号) を取得する。 $O(1)$?scc()
: 強連結成分分解cycle()
: サイクル検出is_connected()
: 連結判定eulerian_walk()
: オイラー路#pragma once
#include <cassert>
#include <optional>
#include <vector>
using namespace std;
#include "../_internal/graph-base.hpp"
#include "./graph.hpp"
namespace asalib {
namespace graph {
class digraph: public _internal::notweighted_graph_base {
public:
digraph(): n_vertex(0), n_edge(0) {}
explicit digraph(size_t n_vertex): n_vertex(n_vertex), n_edge(0) {
adj_list.reserve(n_vertex);
adj_list.resize(n_vertex);
adj_list_rev.reserve(n_vertex);
adj_list_rev.resize(n_vertex);
underlying_graph = graph(n_vertex);
}
void add_edge(size_t from, size_t to) {
assert(0 <= from && from < n_vertex);
assert(0 <= to && to < n_vertex);
adj_list[from].push_back({ to, n_edge });
adj_list_rev[to].push_back({ from, n_edge });
edge_list.push_back({ from, to });
underlying_graph.add_edge(from, to);
++n_edge;
}
// (from, to)
pair<size_t, size_t> get_edge(size_t edgeId) const {
assert(0 <= edgeId && edgeId < n_edge);
return edge_list[edgeId];
}
// (to, edgeId)
vector<pair<size_t, size_t>> get_adj(size_t vertex) const {
assert(0 <= vertex && vertex < n_vertex);
return adj_list[vertex];
}
// (from, edgeId)
vector<pair<size_t, size_t>> get_adjrev(size_t vertex) const {
assert(0 <= vertex && vertex < n_vertex);
return adj_list_rev[vertex];
}
// ---------- prototype ---------- //
vector<vector<size_t>> scc() const;
optional<pair<vector<size_t>, vector<size_t>>> cycle() const;
bool is_connected() const;
optional<pair<vector<size_t>, vector<size_t>>> eulerian_walk() const;
private:
size_t n_vertex, n_edge;
asalib::_internal::adjlist_t adj_list, adj_list_rev;
asalib::_internal::edgelist_t edge_list;
graph underlying_graph;
};
} // namespace graph
} // namespace asalib
#line 2 "library/data-structure/digraph.hpp"
#include <cassert>
#include <optional>
#include <vector>
using namespace std;
#line 2 "library/_internal/graph-base.hpp"
#include <concepts>
#include <type_traits>
using namespace std;
namespace asalib {
namespace _internal {
class graph_base {};
class notweighted_graph_base: public graph_base {};
class weighted_graph_base: public graph_base {};
template<typename T>
concept is_graph = is_base_of_v<graph_base, T>;
template<typename T>
concept notweighted_graph = is_base_of_v<notweighted_graph_base, T>;
template<typename T>
concept weighted_graph = is_base_of_v<weighted_graph_base, T>;
using adjlist_t = vector<vector<pair<size_t, size_t>>>;
using edgelist_t = vector<pair<size_t, size_t>>;
} // namespace _internal
} // namespace asalib
#line 2 "library/data-structure/graph.hpp"
#line 6 "library/data-structure/graph.hpp"
using namespace std;
#line 9 "library/data-structure/graph.hpp"
namespace asalib {
namespace graph {
class graph: public _internal::notweighted_graph_base {
public:
graph(): n_vertex(0), n_edge(0) {}
explicit graph(size_t n_vertex): n_vertex(n_vertex), n_edge(0) {
adj_list.reserve(n_vertex);
adj_list.resize(n_vertex);
}
void add_edge(size_t v1, size_t v2) {
assert(0 <= v1 && v1 < n_vertex);
assert(0 <= v2 && v2 < n_vertex);
adj_list[v1].push_back({ v2, n_edge });
adj_list[v2].push_back({ v1, n_edge });
edge_list.push_back({ v1, v2 });
++n_edge;
}
// (v1, v2)
pair<size_t, size_t> get_edge(size_t edgeId) const {
assert(0 <= edgeId && edgeId < n_edge);
return edge_list[edgeId];
}
// (v2, edgeId)
vector<pair<size_t, size_t>> get_adj(size_t vertex) const {
assert(0 <= vertex && vertex < n_vertex);
return adj_list[vertex];
}
// ---------- prototype ---------- //
optional<pair<vector<size_t>, vector<size_t>>> cycle() const;
bool is_connected() const;
optional<pair<vector<size_t>, vector<size_t>>> eulerian_walk() const;
private:
size_t n_vertex, n_edge;
asalib::_internal::adjlist_t adj_list;
asalib::_internal::edgelist_t edge_list;
};
} // namespace graph
} // namespace asalib
#line 10 "library/data-structure/digraph.hpp"
namespace asalib {
namespace graph {
class digraph: public _internal::notweighted_graph_base {
public:
digraph(): n_vertex(0), n_edge(0) {}
explicit digraph(size_t n_vertex): n_vertex(n_vertex), n_edge(0) {
adj_list.reserve(n_vertex);
adj_list.resize(n_vertex);
adj_list_rev.reserve(n_vertex);
adj_list_rev.resize(n_vertex);
underlying_graph = graph(n_vertex);
}
void add_edge(size_t from, size_t to) {
assert(0 <= from && from < n_vertex);
assert(0 <= to && to < n_vertex);
adj_list[from].push_back({ to, n_edge });
adj_list_rev[to].push_back({ from, n_edge });
edge_list.push_back({ from, to });
underlying_graph.add_edge(from, to);
++n_edge;
}
// (from, to)
pair<size_t, size_t> get_edge(size_t edgeId) const {
assert(0 <= edgeId && edgeId < n_edge);
return edge_list[edgeId];
}
// (to, edgeId)
vector<pair<size_t, size_t>> get_adj(size_t vertex) const {
assert(0 <= vertex && vertex < n_vertex);
return adj_list[vertex];
}
// (from, edgeId)
vector<pair<size_t, size_t>> get_adjrev(size_t vertex) const {
assert(0 <= vertex && vertex < n_vertex);
return adj_list_rev[vertex];
}
// ---------- prototype ---------- //
vector<vector<size_t>> scc() const;
optional<pair<vector<size_t>, vector<size_t>>> cycle() const;
bool is_connected() const;
optional<pair<vector<size_t>, vector<size_t>>> eulerian_walk() const;
private:
size_t n_vertex, n_edge;
asalib::_internal::adjlist_t adj_list, adj_list_rev;
asalib::_internal::edgelist_t edge_list;
graph underlying_graph;
};
} // namespace graph
} // namespace asalib