summaryrefslogtreecommitdiff
path: root/content/graph/cycleCounting.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'content/graph/cycleCounting.cpp')
-rw-r--r--content/graph/cycleCounting.cpp64
1 files changed, 64 insertions, 0 deletions
diff --git a/content/graph/cycleCounting.cpp b/content/graph/cycleCounting.cpp
new file mode 100644
index 0000000..6a299ee
--- /dev/null
+++ b/content/graph/cycleCounting.cpp
@@ -0,0 +1,64 @@
+constexpr int maxEdges = 128;
+using cycle = bitset<maxEdges>;
+struct cycles {
+ vector<vector<pair<int, int>>> adj;
+ vector<bool> seen;
+ vector<cycle> paths, base;
+ vector<pair<int, int>> edges;
+
+ cycles(int n) : adj(n), seen(n), paths(n) {}
+
+ void addEdge(int u, int v) {
+ adj[u].push_back({v, sz(edges)});
+ adj[v].push_back({u, sz(edges)});
+ edges.push_back({u, v});
+ }
+
+ void addBase(cycle cur) {
+ for (cycle o : base) {
+ o ^= cur;
+ if (o._Find_first() > cur._Find_first()) cur = o;
+ }
+ if (cur.any()) base.push_back(cur);
+ }
+
+ void findBase(int v, int from = -1, cycle cur = {}) {
+ if (from < 0 && seen[v]) return;
+ if (seen[v]) {
+ addBase(cur ^ paths[v]);
+ } else {
+ seen[v] = true;
+ paths[v] = cur;
+ for (auto [u, id] : adj[v]) {
+ if (u == from) continue;
+ cur[id].flip();
+ findBase(u, v, cur);
+ cur[id].flip();
+ }}}
+
+ bool isCycle(cycle cur) {//cycle must be constrcuted from base
+ if (cur.none()) return false;
+ init(sz(adj)); // union find @\sourceref{datastructures/unionFind.cpp}@
+ for (int i = 0; i < sz(edges); i++) {
+ if (cur[i]) {
+ cur[i] = false;
+ if (findSet(edges[i].first) ==
+ findSet(edges[i].second)) break;
+ unionSets(edges[i].first, edges[i].second);
+ }}
+ return cur.none();
+ }
+
+ int count() {
+ for (int i = 0; i < sz(adj); i++) findBase(i);
+ assert(sz(base) < 30);
+ int res = 0;
+ for (int i = 1; i < (1 << sz(base)); i++) {
+ cycle cur;
+ for (int j = 0; j < sz(base); j++)
+ if (((i >> j) & 1) != 0) cur ^= base[j];
+ if (isCycle(cur)) res++;
+ }
+ return res;
+ }
+};