二分图的最佳匹配(KM 算法)

原创 2007年09月19日 05:21:00
找了标程研究了一下,修改成自己的风格了,贴上来。
 
引用:
KM算法是通过给每个顶点一个标号(叫做顶标)来把求最大权匹配的问题转化为求完备匹配的问题的。设顶点Xi的顶标为A[i],顶点Yi的顶标为B [i],顶点Xi与Yj之间的边权为w[i,j]。在算法执行过程中的任一时刻,对于任一条边(i,j),A[i]+B[j]>=w[i,j]始终 成立。KM算法的正确性基于以下定理:
  若由二分图中所有满足A[i]+B[j]=w[i,j]的边(i,j)构成的子图(称做相等子图)有完备匹配,那么这个完备匹配就是二分图的最大权匹配。
  这个定理是显然的。因为对于二分图的任意一个匹配,如果它包含于相等子图,那么它的边权和等于所有顶点的顶标和;如果它有的边不包含于相等子图,那么它的边权和小于所有顶点的顶标和。所以相等子图的完备匹配一定是二分图的最大权匹配。
  初始时为了使A[i]+B[j]>=w[i,j]恒成立,令A[i]为所有与顶点Xi关联的边的最大权,B[j]=0。如果当前的相等子图没有完备匹配,就按下面的方法修改顶标以使扩大相等子图,直到相等子图具有完备匹配为止。
  我们求当前相等子图的完备匹配失败了,是因为对于某个X顶点,我们找不到一条从它出发的交错路。这时我们获得了一棵交错树,它的叶子结点全部是X顶点。现在我们把交错树中X顶点的顶标全都减小某个值d,Y顶点的顶标全都增加同一个值d,那么我们会发现:
两端都在交错树中的边(i,j),A[i]+B[j]的值没有变化。也就是说,它原来属于相等子图,现在仍属于相等子图。
两端都不在交错树中的边(i,j),A[i]和B[j]都没有变化。也就是说,它原来属于(或不属于)相等子图,现在仍属于(或不属于)相等子图。
X端不在交错树中,Y端在交错树中的边(i,j),它的A[i]+B[j]的值有所增大。它原来不属于相等子图,现在仍不属于相等子图。
X端在交错树中,Y端不在交错树中的边(i,j),它的A[i]+B[j]的值有所减小。也就说,它原来不属于相等子图,现在可能进入了相等子图,因而使相等子图得到了扩大。
  现在的问题就是求d值了。为了使A[i]+B[j]>=w[i,j]始终成立,且至少有一条边进入相等子图,d应该等于min{A[i]+B[j]-w[i,j]|Xi在交错树中,Yi不在交错树中}。
  以上就是KM算法的基本思路。但是朴素的实现方法,时间复杂度为O(n4)——需要找O(n)次增广路,每次增广最多需要修改O(n)次顶 标,每次修改顶标时由于要枚举边来求d值,复杂度为O(n2)。实际上KM算法的复杂度是可以做到O(n3)的。我们给每个Y顶点一个“松弛量”函数 slack,每次开始找增广路时初始化为无穷大。在寻找增广路的过程中,检查边(i,j)时,如果它不在相等子图中,则让slack[j]变成原值与A [i]+B[j]-w[i,j]的较小值。这样,在修改顶标时,取所有不在交错树中的Y顶点的slack值中的最小值作为d值即可。但还要注意一点:修改 顶标后,要把所有的slack值都减去d。

#include <cstdio>
#include 
<memory.h>
#include 
<algorithm>    // 使用其中的 min 函数
using namespace std;

const int MAX = 1024;

int n;                // X 的大小
int weight [MAX] [MAX];        // X 到 Y 的映射(权重)
int lx [MAX], ly [MAX];        // 标号
bool sx [MAX], sy [MAX];    // 是否被搜索过
int match [MAX];        // Y(i) 与 X(match [i]) 匹配

// 初始化权重
void init (int size);
// 从 X(u) 寻找增广道路,找到则返回 true
bool path (int u);
// 参数 maxsum 为 true ,返回最大权匹配,否则最小权匹配
int bestmatch (bool maxsum = true);

void init (int size)
{
    
// 根据实际情况,添加代码以初始化
    n = size;
    
for (int i = 0; i < n; i ++)
        
for (int j = 0; j < n; j ++)
            scanf (
"%d"&weight [i] [j]);
}


bool path (int u)
{
    sx [u] 
= true;
    
for (int v = 0; v < n; v ++)
        
if (!sy [v] && lx[u] + ly [v] == weight [u] [v])
            {
            sy [v] 
= true;
            
if (match [v] == -1 || path (match [v]))
                {
                match [v] 
= u;
                
return true;
                }
            }
    
return false;
}

int bestmatch (bool maxsum)
{
    
int i, j;
    
if (!maxsum)
        {
        
for (i = 0; i < n; i ++)
            
for (j = 0; j < n; j ++)
                weight [i] [j] 
= -weight [i] [j];
        }

    
// 初始化标号
    for (i = 0; i < n; i ++)
        {
        lx [i] 
= -0x1FFFFFFF;
        ly [i] 
= 0;
        
for (j = 0; j < n; j ++)
            
if (lx [i] < weight [i] [j])
                lx [i] 
= weight [i] [j];
        }

    memset (match, 
-1sizeof (match));
    
for (int u = 0; u < n; u ++)
        
while (1)
            {
            memset (sx, 
0sizeof (sx));
            memset (sy, 
0sizeof (sy));
            
if (path (u))
                
break;

            
// 修改标号
            int dx = 0x7FFFFFFF;
            
for (i = 0; i < n; i ++)
                
if (sx [i])
                    
for (j = 0; j < n; j ++)
                        
if(!sy [j])
                            dx 
= min (lx[i] + ly [j] - weight [i] [j], dx);
            
for (i = 0; i < n; i ++)
                {
                
if (sx [i])
                    lx [i] 
-= dx;
                
if (sy [i])
                    ly [i] 
+= dx;
                }
            }

    
int sum = 0;
    
for (i = 0; i < n; i ++)
        sum 
+= weight [match [i]] [i];

    
if (!maxsum)
        {
        sum 
= -sum;
        
for (i = 0; i < n; i ++)
            
for (j = 0; j < n; j ++)
                weight [i] [j] 
= -weight [i] [j];         // 如果需要保持 weight [ ] [ ] 原来的值,这里需要将其还原
        }
    
return sum;
}


int main()
{
    
int n;
    scanf (
"%d"&n);
    init (n);
    
int cost = bestmatch (true);

    printf (
"%d ", cost);
    
for (int i = 0; i < n; i ++)
        {
        printf (
"Y %d -> X %d ", i, match [i]);
        }

    
return 0;


/*
5
3 4 6 4 9
6 4 5 3 8
7 5 3 4 2
6 3 2 2 5
8 4 5 4 7
//执行bestmatch (true) ,结果为 29
*/

/*
5
7 6 4 6 1
4 6 5 7 2
3 5 7 6 8
4 7 8 8 5
2 6 5 6 3
//执行 bestmatch (false) ,结果为 21
*/

这个实现和图论书上描述的有所不同,这个和匈牙利算法方法上是一样的(不断地寻找增广道路。。),而不是像书上在过程中调用匈牙利算法。。

相关文章推荐

二分图最佳匹配 KM算法 Hdu2255奔小康赚大钱 + Poj 3565 Ants

理论:http://blog.sina.com.cn/s/blog_691ce2b701016reh.html http://philoscience.iteye.com/blog/1754498 ...

HDU2255 奔小康赚大钱 【二分图最佳匹配·KM算法】

奔小康赚大钱 Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Tot...

poj2516 Minimum Cost 二分图最佳匹配,km算法

Minimum Cost Time Limit: 4000MS   Memory Limit: 65536K Total Submissions: 17135   ...

KM算法 二分图的最佳匹配

引用: KM算法是通过给每个顶点一个标号(叫做顶标)来把求最大权匹配的问题转化为求完备匹配的问题的。设顶点Xi的顶标为A[i],顶点Yi的顶标为B [i],顶点Xi与Yj之间的边权为w[i,j]。在...
  • fp_hzq
  • fp_hzq
  • 2011年03月27日 18:43
  • 612

带权二分图的最佳匹配(KM算法)

还是没看懂一般图都是最大匹配问题。。怪我太笨了哎~ 先来个看明白了的KM算法——寻找带权二分图的最佳匹配方法 一般对KM算法的描述,基本上可以概括成以下几个步骤: (1) 初始化可行标杆 (...
  • x_y_q_
  • x_y_q_
  • 2016年07月16日 23:17
  • 3173

二分图的最佳匹配(KM 算法)

KM算法求最小权二分匹配,模板题,构图很简单,直接把人当作左边的点,房子当作右边的点, 两者之间的曼哈顿距离当作权值即可。第一次搞带权二分匹配的题,就是用KM算法求最小权的时候要加个处,由于KM求的...
  • weiqubo
  • weiqubo
  • 2011年11月29日 10:44
  • 8041

二分图匹配之最佳匹配 km算法详解

今天也大致学了下KM算法,用于求二分图匹配的最佳匹配。 何为最佳?我们能用匈牙利算法对二分图进行最大匹配,但匹配的方式不唯一,如果我们假设每条边有权值,那么一定会存在一个最大权值的匹配情况,但对于K...

HDU 2255奔小康赚大钱 二分图最佳匹配KM

Description 传说在遥远的地方有一个非常富裕的村落,有一天,村长决定进行制度改革:重新分配房子。  这可是一件大事,关系到人民的住房问题啊。村里共有n间房间,刚好有n家老百姓,考虑到...

ZOJ-3933-Team Formation【二分图最佳匹配】【KM】

3933-Team Formation

二分图最佳完美匹配-KM算法

二分图最佳完美匹配-KM算法总结。
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:二分图的最佳匹配(KM 算法)
举报原因:
原因补充:

(最多只允许输入30个字)