匈牙利算法用来求二分图最大匹配
km算法求带权的最大匹配
参考http://wenku.baidu.com/view/be01d3e84afe04a1b071de54.html
km算法
#include <iostream>
#include <cstdio>
#include <memory.h>
#include <algorithm>
using namespace std;
#define MAX 100
int n;
int weight[MAX][MAX]; //权重
int lx[MAX],ly[MAX]; //定点标号
bool sx[MAX],sy[MAX]; //记录寻找增广路时点集x,y里的点是否搜索过
int match[MAX]; //match[i]记录y[i]与x[match[i]]相对应
bool search_path(int u) { //给x[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 || search_path(match[v])){
match[v]=u;
return true;
}
}
}
return false;
}
int Kuhn_Munkras(bool max_weight){
if(!max_weight){ //如果求最小匹配,则要将边权取反
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
weight[i][j]=-weight[i][j];
}
//初始化顶标,lx[i]设置为max(weight[i][j] | j=0,..,n-1 ), ly[i]=0;
for(int i=0;i<n;i++){
ly[i]=0;
lx[i]=-0x7fffffff;
for(int j=0;j<n;j++)
if(lx[i]<weight[i][j])
lx[i]=weight[i][j];
}
memset(match,-1,sizeof(match));
//不断修改顶标,直到找到完备匹配或完美匹配
for(int u=0;u<n;u++){ //为x里的每一个点找匹配
while(1){
memset(sx,0,sizeof(sx));
memset(sy,0,sizeof(sy));
if(search_path(u)) //x[u]在相等子图找到了匹配,继续为下一个点找匹配
break;
//如果在相等子图里没有找到匹配,就修改顶标,直到找到匹配为止
//首先找到修改顶标时的增量inc, min(lx[i] + ly [i] - weight[i][j],inc);,lx[i]为搜索过的点,ly[i]是未搜索过的点,因为现在是要给u找匹配,所以只需要修改找的过程中搜索过的点,增加有可能对u有帮助的边
int inc=0x7fffffff;
for(int i=0;i<n;i++)
if(sx[i])
for(int j=0;j<n;j++)
if(!sy[j]&&((lx[i] + ly [j] - weight[i][j] )<inc))
inc = lx[i] + ly [j] - weight[i][j] ;
//找到增量后修改顶标,因为sx[i]与sy[j]都为真,则必然符合lx[i] + ly [j] =weight[i][j],然后将lx[i]减inc,ly[j]加inc不会改变等式,但是原来lx[i] + ly [j] !=weight[i][j]即sx[i]与sy[j]最多一个为真,lx[i] + ly [j] 就会发生改变,从而符合等式,边也就加入到相等子图中
if(inc==0) cout<<"fuck!"<<endl;
for(int i=0;i<n;i++){
if(sx[i]) //
lx[i]-=inc;
if(sy[i])
ly[i]+=inc;
}
}
}
int sum=0;
for(int i=0;i<n;i++)
if(match[i]>=0)
sum+=weight[match[i]][i];
if(!max_weight)
sum=-sum;
return sum;
}
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
scanf("%d",&weight[i][j]);
printf("%d\n",Kuhn_Munkras(1));
system("pause");
return 0;
}