blob: cd8b83255da178ceefa4f5ff343d5f74432ac66a (
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
|
function order = topological_sort(A)
% TOPOLOGICAL_SORT Return the nodes in topological order (parents before children).
% order = topological_sort(adj_mat)
n = length(A);
indeg = zeros(1,n);
zero_indeg = []; % a stack of nodes with no parents
for i=1:n
indeg(i) = length(parents(A,i));
if indeg(i)==0
zero_indeg = [i zero_indeg];
end
end
t=1;
order = zeros(1,n);
while ~isempty(zero_indeg)
v = zero_indeg(1); % pop v
zero_indeg = zero_indeg(2:end);
order(t) = v;
t = t + 1;
cs = children(A, v);
for j=1:length(cs)
c = cs(j);
indeg(c) = indeg(c) - 1;
if indeg(c) == 0
zero_indeg = [c zero_indeg]; % push c
end
end
end
|