既然学过了匈牙利算法,那就可以解决不带边权的二分图最大匹配,但如果带边权呢?
那就要用到KM算法了。
KM算法,就是在求最大匹配的时候在加上一个边权,就是求带权二分图的最优匹配,这样的话匈牙利算法就不能很好的解释了,这就用到了KM算法。
KM算法:
每次都帮一个顶点匹配最大权重边,利用匈牙利算法完成最大匹配,最终我们完成的就是最优匹配!
KM算法流程:
- 初始化可行顶标的值
- 用匈牙利算法寻找完备匹配
- 若未找到完备匹配则修改可行顶标的值
- 重复(2)(3)直到找到相等子图的完备匹配为止。
(表示刚开始学的时候是一点一点跟着视频匹配的,,匹配了三四遍,,/(ㄒoㄒ)/~~)
如果刚开始学的话,这篇大佬的博客模拟的很清楚:传送门
题目:
代码:
#include <stdio.h>
#include <string.h>
using namespace std;
inline int read()
{
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=1;ch=getchar();}
while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
return s*w;
}
const int ocean=0x3fffffff;
const int sea=500;
int n,a[sea][sea],vx[sea],vy[sea],ex[sea],ey[sea],pre[sea],slack[sea];
bool hgry(int x)
{
vx[x]=1;
for(int y=1;y<=n;y++)
{
if(vy[y]) continue;
int cc=ex[x]+ey[y]-a[x][y];
if(cc==0)
{
vy[y]=1;
if(pre[y]==-1||hgry(pre[y]))
{pre[y]=x; return 1;}
}
else if(slack[y]>cc) slack[y]=cc;
}
return 0;
}
int KM()
{
memset(pre,-1,sizeof(pre));
memset(ex,0,sizeof(ex));
memset(ey,0,sizeof(ey));
for(int i=1;i<=n;i++) for(int j=1;j<=n;j++)
if (a[i][j]>ex[i]) ex[i]=a[i][j];
for(int x=1;x<=n;x++)
{
for (int i=1;i<=n;i++) slack[i]=ocean;
while(1)
{
memset(vx,0,sizeof(vx));
memset(vy,0,sizeof(vy));
if(hgry(x)) break;
int cc=ocean;
for(int i=1;i<=n;i++) if(!vy[i]&&cc>slack[i]) cc=slack[i];
for(int i=1;i<=n;i++) if(vx[i]) ex[i]-=cc;
for(int i=1;i<=n;i++) if(vy[i]) ey[i]+=cc;else slack[i]-=cc;
}
}
int ans=0;
for(int i=1;i<=n;i++) if(pre[i]>-1) ans+=a[pre[i]][i];
return ans;
}
int main()
{
while (scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) a[i][j]=read();
printf("%d\n",KM());
}
return 0;
}