struct graph {
struct node {
int u, v, w;
bool operator<(const node &other) const {
return w < other.w;
}
bool operator==(const node &other) const {
return u == other.u && v == other.v && w == other.w;
}
};
vector<vector<node>> e;
int n;
bool directed;
bool weighted = 0;
graph(int V, bool D = 0) {
n = V;
e.resize(n);
directed = D;
}
void add_edge(int u, int v, int w = 1) {
e[u].push_back(node{u, v, w});
}
int spfa(int s, vector<int> &dis, vector<int> &pre) {
dis.resize(n, INF);
pre.resize(n, -1);
vector<bool> inq(n, 0);
vector<int> cnt(n, 0);
queue<int> q;
dis[s] = 0, q.push(s), inq[s] = 1;
while (!q.empty()) {
int u = q.front(); q.pop();
inq[u] = 0;
for (auto edge: e[u]) {
int v = edge.v, w = edge.w;
if (dis[u] != INF && dis[v] > dis[u] + w) {
pre[v] = u;
dis[v] = dis[u] + w;
if (!inq[v]) {
q.push(v), inq[v] = 1, ++cnt[v];
if (cnt[v] >= n) return v;
}
}
}
}
return -1;
}
void dijkstra(int s, vector<int> &dis) {
dis.resize(n, INF);
vector<bool> vis(n, 0);
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
dis[s] = 0, pq.push({0, s});
while (!pq.empty()) {
int u = pq.top().second; pq.pop();
if (vis[u]) continue;
vis[u] = 1;
for (auto edge: e[u]) {
int v = edge.v, w = edge.w;
if (dis[v] > dis[u] + w) {
dis[v] = dis[u] + w;
pq.push({dis[v], v});
}
}
}
}
void floyd(vector<vector<int>> &dis) {
dis.resize(n, vector<int>(n, INF));
rep(i, 0, n) dis[i][i] = 0;
rep(u, 0, n) for (auto edge: e[u]) {
int v = edge.v, w = edge.w;
dis[u][v] = min(dis[u][v], w);
}
rep(k, 0, n) rep(i, 0, n) rep(j, 0, n) {
if (dis[i][k] == INF || dis[k][j] == INF) continue;
dis[i][j] = min(dis[i][j], dis[i][k] + dis[k][j]);
}
}
void graphviz_dump(string filename = "graph.dot") {
ofstream gf;
gf.open(filename);
gf << (directed ? "digraph" : "graph") << " {\n";
gf << " "; rep(i, 0, n) gf << i << " ;"[i==n-1]; gf << endl;
string notation = directed ? " -> " : " -- ";
for (auto es: e) for (auto edge: es) if (edge.w != 1) weighted = 1;
for (auto es: e) {
for (auto edge: es) {
if (!directed && edge.u > edge.v) continue;
gf << " " << edge.u << notation << edge.v << (weighted ? " [label=\"" + to_string(edge.w) + "\"];\n" : ";\n");
}
}
gf << "}\n";
}
};
No Comments