summaryrefslogtreecommitdiff
path: root/graph/scc.cpp
blob: b33616b1eb1e0f6a1b5ea93c2cb3e6ee4e5f3100 (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
int counter, sccCounter, n; //n == number of vertices
vector<bool> visited, inStack;
vector< vector<int> > adjlist;
vector<int> d, low, sccs;
stack<int> s;

void visit(int v) {
	visited[v] = true;
	d[v] = counter;
	low[v] = counter;
	counter++;
	inStack[v] = true;
	s.push(v);
	
	for (int i = 0; i < (int)adjlist[v].size(); i++) {
		int u = adjlist[v][i];
		if (!visited[u]) {
			visit(u);
			low[v] = min(low[v], low[u]);
		} else if (inStack[u]) {
			low[v] = min(low[v], low[u]);
		}
	}
	
	if (d[v] == low[v]) {
		int u;
		do {
			u = s.top();
			s.pop();
			inStack[u] = false;
			sccs[u] = sccCounter;
		} while(u != v);
		sccCounter++;
	}
}

void scc() {
	//read adjlist
	
	visited.clear(); visited.assign(n, false);
	d.clear(); d.resize(n);
	low.clear(); low.resize(n);
	inStack.clear(); inStack.assign(n, false);
	sccs.clear(); sccs.resize(n);
	
	counter = 0;
	sccCounter = 0;
	for (i = 0; i < n; i++) {
		if (!visited[i]) {
			visit(i);
		}
	}
	//sccs has the component for each vertex
}