Merge pull request #361 from hitonanode/block-cut-tree

block cut tree

Author: hitonanode

graph/extended_block_cut_trees.hpp

@@ -0,0 +1,116 @@
+#pragma once
+
+#include <cassert>
+#include <utility>
+#include <vector>
+
+// Construct block cut tree (or forest) from a given graph
+// Complexity: O(N + M), N = |vertices|, M = |edges|
+// based on this idea: https://x.com/noshi91/status/1529858538650374144
+// based on this implementation: https://ssrs-cp.github.io/cp_library/graph/extended_block_cut_tree.hpp.html
+struct extended_block_cut_trees {
+    int N;                            // number of vertices
+    int B;                            // number of blocks
+    std::vector<std::vector<int>> to; // (0, ..., N - 1): vertices, (N, ..., N + B - 1): blocks
+
+    extended_block_cut_trees(int N, const std::vector<std::pair<int, int>> &edges)
+        : N(N), B(0), to(N) {
+        std::vector<std::vector<int>> adj(N);
+        for (auto [u, v] : edges) {
+            if (u != v) adj.at(u).push_back(v), adj.at(v).push_back(u);
+        }
+
+        std::vector<int> dfs_next(N, -1), dist(N, -1), back_cnt(N);
+
+        auto rec1 = [&](auto &&self, int now) -> void {
+            for (int nxt : adj[now]) {
+                if (dist[nxt] == -1) {
+                    dist[nxt] = dist[now] + 1;
+                    dfs_next[now] = nxt;
+                    self(self, nxt);
+                    back_cnt[now] += back_cnt[nxt];
+                } else if (dist[nxt] < dist[now] - 1) {
+                    ++back_cnt[now];
+                    --back_cnt[dfs_next[nxt]];
+                }
+            }
+        };
+
+        for (int i = 0; i < N; ++i) {
+            if (dist[i] == -1) dist[i] = 0, rec1(rec1, i);
+        }
+
+        std::vector<bool> used(N);
+
+        auto rec2 = [&](auto &&self, int now, int current_b) -> void {
+            used[now] = true;
+            bool ok = false;
+
+            for (int nxt : adj[now]) {
+                if (dist[nxt] == dist[now] + 1 and !used[nxt]) {
+                    if (back_cnt[nxt] > 0) {
+                        if (!ok) {
+                            ok = true;
+                            add_edge(now, current_b);
+                        }
+                        self(self, nxt, current_b);
+                    } else {
+                        to.push_back({});
+                        ++B;
+                        add_edge(now, B - 1);
+                        self(self, nxt, B - 1);
+                    }
+                }
+            }
+
+            if (!ok and dist[now] > 0) { add_edge(now, current_b); }
+        };
+
+        for (int i = 0; i < N; ++i) {
+            if (dist[i] == 0) { rec2(rec2, i, B - 1); }
+            if (adj[i].empty()) {
+                to.push_back({});
+                ++B;
+                add_edge(i, B - 1);
+            }
+        }
+    }
+
+    int size() const { return N + B; }
+
+    bool is_articulation_point(int vertex) const {
+        assert(0 <= vertex and vertex < N);
+        return to[vertex].size() > 1;
+    }
+
+    int block_size(int block) const {
+        assert(0 <= block and block < B);
+        return to[N + block].size();
+    }
+
+    const std::vector<int> &block_vertices(int block) const {
+        assert(0 <= block and block < B);
+        return to[N + block];
+    }
+
+    std::vector<std::vector<int>> biconnected_components() const {
+        return std::vector<std::vector<int>>(to.begin() + N, to.end());
+    }
+
+    // first < N (vertices), second >= N (blocks)
+    std::vector<std::pair<int, int>> get_edges() const {
+        std::vector<std::pair<int, int>> edges;
+        for (int i = 0; i < N; ++i) {
+            for (int j : to[i]) edges.emplace_back(i, j);
+        }
+        return edges;
+    }
+
+private:
+    void add_edge(int vertex, int block) {
+        assert(0 <= vertex and vertex < N);
+        assert(0 <= block and block < B);
+        to[vertex].push_back(N + block);
+        to[N + block].push_back(vertex);
+    }
+};

graph/extended_block_cut_trees.md

@@ -0,0 +1,55 @@
+---
+title: Extended block cut tree （Block cut tree の亜種）
+documentation_of: ./extended_block_cut_trees.hpp
+---
+
+与えられた $n$ 頂点 $m$ 辺の無向グラフに対する block cut tree （のようなもの）を構築する．計算量は $O(n + m)$．
+
+詳細な仕様は以下の通り．
+
+- 入力グラフの二重連結成分数を $k$ とすると，頂点数 $n + k$ のグラフ（特に，森）を構築する．
+  - **通常の block cut tree との相違点として，関節点ではない頂点も構築されたグラフに含まれる．**
+- 頂点番号 $0, \ldots, n - 1$ が入力グラフの各頂点， $n, \ldots, n + k - 1$ が入力グラフの各二重連結成分に対応する．
+- 頂点 $v$ が二重連結成分 $i$ に含まれるとき，頂点 $v$ と $n + i$ の間に辺が張られる．
+- このコードにおいて，次数 0 の頂点（孤立点）はそれ単独で二重連結成分と見なされる．
+  - したがって，入力グラフの全ての頂点が一つ以上の二重連結成分に属することになる．
+
+なお，出力されるグラフは以下の性質を満たす．
+
+- 一般に出力されるグラフは森．特に入力グラフが連結ならば出力されるグラフも連結（木）．
+- 入力グラフの頂点 $v$ が関節点であることは，出力されるグラフの頂点 $v$ の次数が $2$ 以上であることと同値．
+
+## 使用方法
+
+```cpp
+int n;  // Num. of vertices
+vector<pair<int, int>> edges;
+
+const extended_block_cut_trees bct(n, edges);
+
+int b = bct.B;  // Num. of blocks
+vector<vector<int>> bct_to = bct.to;
+assert(n + b == (int)bct_to.size());
+
+// 得られた block cut tree を heavy-light decomposition などに使う場合
+HeavyLightDecomposition hld(bct.size());
+for (auto [i, j] : bct.get_edges()) hld.add_edge(i, j);
+```
+
+## 問題例
+
+- [Library Checker: Biconnected Components](https://judge.yosupo.jp/problem/biconnected_components)
+- [No.1326 ふたりのDominator - yukicoder](https://yukicoder.me/problems/no/1326)
+  - 無向グラフにおいて指定された 2 頂点間の任意のパスが含む頂点の集合は， block cut tree でその 2 頂点間のパス上に存在する頂点の集合と一致する．
+- [AtCoder Beginner Contest 318 G - Typical Path Problem](https://atcoder.jp/contests/abc318/tasks/abc318_g)
+- [AtCoder Beginner Contest 334 G - Christmas Color Grid 2](https://atcoder.jp/contests/abc334/tasks/abc334_g)
+- [灘校文化祭コンテスト 2022 Day2 H - Tourist on Graph](https://atcoder.jp/contests/nadafes2022_day2/tasks/nadafes2022_day2_h)
+- [Codeforces Round 606 (Div. 1, based on Technocup 2020 Elimination Round 4) B. Two Fairs - Codeforces](https://codeforces.com/contest/1276/problem/B)
+- [2022-2023 Winter Petrozavodsk Camp, Day 2: GP of ainta I. Visiting Friend - Codeforces](https://codeforces.com/gym/104427/problem/I)
+
+## 参考文献・リンク
+
+- [Block-cut tree - ei1333の日記](https://ei1333.hateblo.jp/entry/2020/03/25/010057)
+- [My Algorithm : kopricky アルゴリズムライブラリ](https://kopricky.github.io/code/GraphDecomposition/articulation.html)
+- [拡張 Block Cut Tree | cp_library](https://ssrs-cp.github.io/cp_library/graph/extended_block_cut_tree.hpp.html)
+- [Xユーザーの熨斗袋さん: 「G = (V, E) に対する Block Cut Tree、G の2-点連結成分全体を C として (V ⊔ C, {{v, c} | v ∈ c}) で定義すると扱いやすかったりする？」 / X](https://x.com/noshi91/status/1529858538650374144)

graph/test/extended_block_cut_trees.test.cpp

@@ -0,0 +1,27 @@
+#define PROBLEM "https://judge.yosupo.jp/problem/biconnected_components"
+#include "../extended_block_cut_trees.hpp"
+
+#include <atcoder/fenwicktree>
+
+#include <iostream>
+
+using namespace std;
+
+int main() {
+    cin.tie(nullptr);
+    ios::sync_with_stdio(false);
+
+    int N, M;
+    cin >> N >> M;
+    vector<pair<int, int>> edges(M);
+    for (auto &[u, v] : edges) cin >> u >> v;
+
+    const extended_block_cut_trees bct(N, edges);
+
+    cout << bct.B << '\n';
+    for (const auto &g : bct.biconnected_components()) {
+        cout << g.size();
+        for (int v : g) cout << ' ' << v;
+        cout << '\n';
+    }
+}

graph/test/extended_block_cut_trees.yuki1326.test.cpp

@@ -0,0 +1,42 @@
+#define PROBLEM "https://yukicoder.me/problems/no/1326"
+#include "../extended_block_cut_trees.hpp"
+#include "../../tree/heavy_light_decomposition.hpp"
+
+#include <atcoder/fenwicktree>
+
+#include <iostream>
+
+using namespace std;
+
+int main() {
+    cin.tie(nullptr);
+    ios::sync_with_stdio(false);
+
+    int N, M;
+    cin >> N >> M;
+    vector<pair<int, int>> edges(M);
+    for (auto &[u, v] : edges) cin >> u >> v, --u, --v;
+
+    const extended_block_cut_trees bct(N, edges);
+
+    HeavyLightDecomposition hld(bct.size());
+    for (auto [i, j] : bct.get_edges()) hld.add_edge(i, j);
+    hld.build();
+
+    atcoder::fenwick_tree<int> fw(hld.V);
+    for (int i = 0; i < N; ++i) fw.add(hld.aligned_id[i], 1);
+
+    int Q;
+    cin >> Q;
+    while (Q--) {
+        int u, v;
+        cin >> u >> v;
+        --u, --v;
+        int ret = 0;
+        if (u != v) {
+            ret = -2;
+            hld.for_each_vertex(u, v, [&](int a, int b) { ret += fw.sum(a, b + 1); });
+        }
+        cout << ret << '\n';
+    }
+}