-
Notifications
You must be signed in to change notification settings - Fork 0
/
hopcroft karp max matching.cpp
92 lines (79 loc) · 1.99 KB
/
hopcroft karp max matching.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include <algorithm>
#include <iostream>
using namespace std;
// you can change these constants to fit your problem needs
const int MAXN1 = 50000;
const int MAXN2 = 50000;
const int MAXM = 150000;
int n1, n2, edges, last[MAXN1], prev[MAXM], head[MAXM];
int matching[MAXN2], dist[MAXN1], Q[MAXN1];
bool used[MAXN1], vis[MAXN1];
void init(int _n1, int _n2) {
n1 = _n1;
n2 = _n2;
edges = 0;
fill(last, last + n1, -1);
}
void addEdge(int u, int v) {
head[edges] = v;
prev[edges] = last[u];
last[u] = edges++;
}
void bfs() {
fill(dist, dist + n1, -1);
int sizeQ = 0;
for (int u = 0; u < n1; ++u) {
if (!used[u]) {
Q[sizeQ++] = u;
dist[u] = 0;
}
}
for (int i = 0; i < sizeQ; i++) {
int u1 = Q[i];
for (int e = last[u1]; e >= 0; e = prev[e]) {
int u2 = matching[head[e]];
if (u2 >= 0 && dist[u2] < 0) {
dist[u2] = dist[u1] + 1;
Q[sizeQ++] = u2;
}
}
}
}
bool dfs(int u1) {
vis[u1] = true;
for (int e = last[u1]; e >= 0; e = prev[e]) {
int v = head[e];
int u2 = matching[v];
if (u2 < 0 || !vis[u2] && dist[u2] == dist[u1] + 1 && dfs(u2)) {
matching[v] = u1;
used[u1] = true;
return true;
}
}
return false;
}
int maxMatching() {
fill(used, used + n1, false);
fill(matching, matching + n2, -1);
for (int res = 0;;) {
bfs();
fill(vis, vis + n1, false);
int f = 0;
for (int u = 0; u < n1; ++u)
if (!used[u] && dfs(u))
++f;
if (!f)
return res;
res += f;
}
}
int main() {
init(2, 2);
addEdge(0, 0);
addEdge(0, 1);
addEdge(1, 1);
cout << (2 == maxMatching()) << endl;
// the actual matching will be stored in matching[] array
// let y be a node from the second part, matching[y] is the node from the first part matched with y.
return 0;
}