题解:
KM算法的模板题,主要学习的两个博客为:
第一个是看他解说KM算法:
https://www.cnblogs.com/Lanly/p/6291214.html
第二个是看他的代码和模拟:
https://www.cnblogs.com/wenruo/p/5264235.html
不多说了,我感觉我来说还不如你们看上面两个更好。
/*KM算法
O(nx*nx*ny)
求最大权匹配
若求最小权匹配,可将权值取相反数,结果再取相反数。
*/
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
const int MAXN=310;
int link[MAXN][MAXN];
int lx[MAXN],ly[MAXN];//x,y中的顶标
int slack[MAXN];
bool visx[MAXN],visy[MAXN];
int cx[MAXN],cy[MAXN];
int nx,ny,n;
bool dfs(int u)
{
visx[u]=true;
for(int v=1;v<=ny;v++)
{
if(visy[v])
continue;
int tmp=lx[u]+ly[v]-link[u][v];
if(tmp==0)
{
visy[v]=true;
if(cy[v]==-1||dfs(cy[v]))
{
cx[u]=v;
cy[v]=u;
return true;
}
}
else
slack[v]=min(slack[v],tmp);
}
return false;
}
int KM()
{
memset(cy,-1,sizeof(cy));
memset(cx,-1,sizeof(cx));
memset(ly,0,sizeof(ly));
for(int i=1;i<=nx;i++)
{
lx[i]=-INF;
for(int j=1;j<=ny;j++)
if(link[i][j]>lx[i])
lx[i]=link[i][j];
}
for(int i=1;i<=nx;i++)
{
// memset(slack,INF,sizeof(slack));
fill(slack+1,slack+ny+1,INF); //黑科技,fill可以赋任何值,而memst只能赋0,-1
// for(int j=1;j<=ny;j++)
// slack[j]=INF;
while(1)
{
memset(visx,false,sizeof(visx));
memset(visy,false,sizeof(visy));
if(dfs(i)) break;
int d=INF;
for(int j=1;j<=ny;j++)
if(!visy[j])
d=min(d,slack[j]);
for(int j=1;j<=nx;j++)
if(visx[j])
lx[j]-=d;
for(int j=1;j<=ny;j++)
{
if(visy[j]) ly[j]+=d;
else slack[j]-=d;
}
}
}
int res=0;
for(int i=1;i<=ny;i++)
if(cy[i]!=-1)
res+=link[cy[i]][i];
return res;
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",&link[i][j]);
nx=ny=n;
int ans=KM();
printf("%d\n",ans);
}
return 0;
}