summaryrefslogtreecommitdiff
path: root/content/graph/cycleCounting.cpp
blob: deac71edbd700f227f0436ea34f7a2f63d12ace6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
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, ssize(edges)});
		adj[v].push_back({u, ssize(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 constructed from base
		if (cur.none()) return false;
		init(ssize(adj)); // union find @\sourceref{datastructures/unionFind.cpp}@
		for (int i = 0; i < ssize(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 < ssize(adj); i++) findBase(i);
		assert(ssize(base) < 30);
		int res = 0;
		for (int i = 1; i < (1 << ssize(base)); i++) {
			cycle cur;
			for (int j = 0; j < ssize(base); j++)
				if (((i >> j) & 1) != 0) cur ^= base[j];
			if (isCycle(cur)) res++;
		}
		return res;
	}
};