KM算法 二分图的最佳匹配

 

引用：

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 <stdio.h> #include <memory.h> #include <algorithm> 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]) 匹配 inline void init()// 初始化权重 { // 根据实际情况，添加代码以初始化 for(int i=0; i<n; ++i) for(int j=0; j<n; ++j) scanf("%d",&weight[i][j]); } bool path(int u)//从 X(u) 寻找增广道路，找到则返回 true { sx[u]=1; for(int v=0; v<n; ++v) if(!sy[v] && lx[u]+ly[v]==weight[u][v]) { sy[v]=1; if(match[v]==-1||path(match[v])) { match[v]=u; return 1; } } return 0; } int bestmatch(bool maxsum)// 参数 maxsum 为 true ，返回最大权匹配，否则最小权匹配 { 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)lx[i]=max(lx[i],weight[i][j]); } memset(match,-1,sizeof(match)); for(int u=0; u<n; ++u) while(1) { memset(sx,0,sizeof(sx)); memset(sy,0,sizeof(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() { scanf("%d",&n); init(); int cost=bestmatch(1); printf("%d/n",cost); 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 (1) ，结果为 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 (0) ，结果为 21

*/

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

 新增模板：

#include <stdio.h> const int MAX = 51; const int MAXX= 151; int nx,ny,slack,temp;//左部图nx,右部图ny,slack松弛值 int w[MAXX][MAXX],lx [MAXX], ly [MAXX],match [MAXX],can[MAXX];//match ,can匹配记录,w边权值lx ly可行顶标 bool sx [MAXX], sy [MAXX],flag; inline int max(int a,int b){return(a>b?a:b);} inline int min(int a,int b){return(a<b?a:b);} bool path(int u) { sx[u]=1; for(int v = 1;v <=ny;++v) { if(sy[v]) continue; temp = w[u][v] - lx[u] - ly[v]; if(temp)slack =min(slack,temp);//更新松弛值 else { sy[v] = 1; if(!match[v] || path(match[v])) { can[u]=v;//记录匹配状态，can可省略 match[v] = u; return 1; } } } return 0; } int KM()//此模板为最小权匹配，传递正权值即可,若要改为最大权匹配，需要修改的参数见注释 { int i,j; for(i=1;i<=nx;++i) { lx[i]=0x7FFFFFFF;//lx[x] = -0x7FFFFFFF; for(j=1;j<=ny;++j)match[j]=0,ly[j]=0,lx[i]=min(lx[i],w[i][j]);lx[i]=max(lx[i],w[i][j]); } for(int u=1;u <=nx;++u) while(1) { for(i=1;i<=nx;++i)sx[i]=0; for(i=1;i<=ny;++i)sy[i]=0; slack=0x7FFFFFFF; if(path(u)) break; for(i=1;i<=nx;++i) if(sx[i])lx[i]+=slack;//lx[x] -= slack; for(j=1;j<=ny;++j) if(sy[j])ly[j]-=slack;//ly[y] += slack; } int sum=0; for(i=1; i<=nx; ++i)sum+=w[i][can[i]]; return sum; } inline int in() { char ch; int a=0; while(!(((ch=getchar())>='0')&&(ch<='9'))); a*=10;a+=ch-'0'; while(((ch=getchar())>='0')&&(ch<='9'))a*=10,a+=ch-'0'; return a; } void out(int a) { if(a>=10)out(a/10); putchar(a%10+'0'); } int main() { int i,j; while(nx=in(),ny=in()) { for(i=1; i<=nx; ++i) for(j=1; j<=ny; ++j)w[i][j]=in(); out(KM()); putchar('/n'); } return 0; }