-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathGetCircles.cpp
70 lines (63 loc) · 1.77 KB
/
GetCircles.cpp
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
65
66
67
68
69
// 有向图找出所有环
auto get_circles([&]() {
vector<int> st(n);
vector<int> v;
vector<vector<int>> res;
function<void(int)> dfs([&](int u) {
st[u] = 1;
v.push_back(u);
for(auto x : adj[u]) {
if(!st[x]) dfs(x);
else {
auto it = find(begin(v), end(v), x);
if(it != v.end()) {
vector<int> tmp;
for(auto i = it; i != v.end(); i ++) {
tmp.push_back(*i);
}
res.push_back(tmp);
}
}
}
v.pop_back();
});
for(int i = 0; i < n; i ++) {
if(!st[i]) dfs(i);
}
return res;
});
// 无向图找出所有环
auto get_circles([&]() {
vector<int> st(n, 0);
vector<int> v;
set<vector<int>> uniqueCycles; // 使用集合来避免重复的环
vector<vector<int>> res;
function<void(int, int)> dfs = [&](int u, int parent) {
st[u] = 1;
v.push_back(u);
for(auto [x, _] : adj[u]) {
if(x == parent) continue; // 跳过父节点
if(!st[x]) {
dfs(x, u);
} else {
auto it = find(begin(v), end(v), x);
if(it != v.end()) {
vector<int> cycle(it, end(v));
sort(begin(cycle), end(cycle)); // 对环进行排序
uniqueCycles.insert(cycle); // 添加到集合中以去重
}
}
}
v.pop_back();
st[u] = 0; // 重置访问标记
};
for(int i = 0; i < n; i++) {
if(!st[i]) {
dfs(i, -1);
}
}
for(const auto& cycle : uniqueCycles) {
res.push_back(cycle);
}
return res;
});