匈牙利算法用于求解无权二分图(unweighted bipartite graph)的最大匹配(maximum matching)问题
二分图
简单来说,有两个点集$U$和$V$ ,集合内部没有边相连,集合之间有边相连,如果存在这样的划分,则此图为一个二分图。二分图的一个等价定义是:不含有「含奇数条边的环」的图。图 1 是一个二分图。为了清晰,我们以后都把它画成图 2 的形式。
匹配
在图论中,「匹配」(matching)是一个边的集合,其中任意两条边都没有公共顶点。例如,图 3、图 4 中红色的边就是图 2 的匹配。
最大匹配
一个图所有匹配中,所含匹配边数最多的匹配,称为这个图的最大匹配。图 4 是一个最大匹配,它包含 4 条匹配边。
匈牙利算法解决的问题背景:如果一对男女互有好感,那么你就可以把这一对撮合在一起,现在,你拥有的大概就是下面这样一张关系图,每一条连线都表示互有好感。
本着救人一命,胜造七级浮屠的原则,你想要尽可能地撮合更多的情侣,匈牙利算法的工作模式会教你这样做:
- 先试着给1号男生找妹子,发现第一个和他相连的1号女生还名花无主,所以!连上一条蓝线
- 接着给2号男生找妹子,发现第一个和他相连的2号女生名花无主,连上一条蓝线
- 接下来是3号男生,很遗憾1号女生已经有主了,怎么办呢?我们试着给之前1号女生匹配的男生(也就是1号男生)另外分配一个妹子。(黄色表示这条边被临时拆掉)
与1号男生相连的第二个女生是2号女生,但是2号女生也有主了,怎么办呢?我们再试着给2号女生的原配,也就是2号男生,重新找个妹子(注意这个步骤和上面是一样的,这是一个递归的过程)
此时发现2号男生还能找到3号女生,那么之前的问题迎刃而解了,回溯回去
2号男生可以找3号妹子
1号男生可以找2号妹子了
3号男生可以找1号妹子
最后是4号男生,很遗憾,按照第三步的节奏没法给4号男生腾出来一个妹子,我们实在是无能为力了。
HDOJ 2063 过山车
就是一个二分图最大匹配模板题,学完之后立刻巩固一下
import java.util.Arrays;
import java.util.Scanner;
public class Main {
static int[][] map;
static int n, m;
public static void main(String[] args) {
Scanner cin = new Scanner(System.in);
while (cin.hasNext()) {
int t = cin.nextInt();
if (t == 0)
break;
n = cin.nextInt();
m = cin.nextInt();
map = new int[n + 1][m + 1];
for (int i = 0; i < t; i++)
map[cin.nextInt()][cin.nextInt()] = 1; // 有向边
int count = 0; // 最大匹配数
int[] mc = new int[m + 1]; // mc[i] = j 表示i号男生所连的女生是j号
Arrays.fill(mc, -1); // 初始时所有女生都没有连
for (int i = 1; i <= n; i++) {
boolean[] vis = new boolean[m + 1]; // vis[i] = true 表示i号男生已经被匹配了
if (dfs(i, vis, mc))
count++;
}
System.out.println(count);
}
}
private static boolean dfs(int start, boolean[] vis, int[] mc) {
for (int i = 1; i <= m; i++) { // 枚举男生集
if (!vis[i] && map[start][i] == 1) { // 如果这个男生没被匹配并且和当前的start有边相连
vis[i] = true; // 将这个男生标记为匹配过
if (mc[i] == -1 || dfs(mc[i], vis, mc)) { // 这个男生没有和女生匹配 || 这个男生所连的女生还有别的选择,就把这个男生"腾"出来
mc[i] = start;
return true;
}
}
}
return false;
}
}
FAQ
Q:为什么题意明明应该是无向图,可是代码的写法是有向图?
A:好问题,其实仔细思考就会发现,二分图求最大匹配的过程中,只用存集合$U$到集合$V$的边,$V$到$U$不需要存,从整个算法思路来看,我们只需要以$U$集合的点作为起始,去往$V$集合。但其实我们也可以从$V$集合回到$U$,利用的就是mc
(memeory connected)数组。如果还不理解请画图仔细体会整个算法流程
Q:还是不太理解if (mc[i] == -1 || dfs(mc[i], vis, mc))
这段代码
A:我们说,集合$U$通过它的边,可以走到集合$V$中的某些点,那么如何判断这些点是否可以与其匹配呢?有两种情况是可以匹配的,首先第一,集合$V$的这个点并没有被匹配,那么我就可以匹配,其次,如果$V$中的这个点(假设这个点是$a$)被匹配了,那么我就看看是谁匹配的$a$,也就是通过$a$反向寻找集合$U$中的点$b$(通过mc
数组),看看$b$还有没有别的匹配选择,把$a$让出来,而这个步骤是个递归的过程。最后一点要注意的是,||
是短路运算,假如条件1成立了就不会执行条件2。
拓展阅读
详细的关于匈牙利算法的原理可以看这篇文章
哈哈,一看就是java代码,太亲切了@(哈哈)
这个代码应该是女生匹配男生吧