http://acm.hdu.edu.cn/status.php
题意:给出村民i与房子j之间的权值,求完美匹配情况下的最大权值。
***还未完全理解km算法,只是找了一个很好的模板,具体理解后还会整理。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
const int inf=0x3f3f3f3f;
int n,m;
int w[505][505];
int slack[505];
int lx[505],ly[505];
int girl[1005];
bool visx[1005],visy[1005];
int dfs(int x)
{
visx[x]=1;
for(int i=1; i<=n; i++)
{
if(!visy[i])
{
int t=lx[x]+ly[i]-w[x][i];
if(t==0) //判断该边是否被加入二分子图中
{
visy[i]=1;
if(girl[i]==-1||dfs(girl[i]))
{
girl[i]=x;
return 1;
}
}
else if(slack[i]>t) //更新y中未被加入的最大边
{
slack[i]=t;
}
}
}
return 0;
}
int km()
{
memset(girl,-1,sizeof(girl));
memset(ly,0,sizeof(ly));
for (int i = 1; i <= n; i ++)
{
lx[i]=-inf;
for (int j = 1; j <= n; j ++) //初始化最大边
if (w[i][j] > lx[i])
lx[i] = w[i][j];
}
for(int x=1; x<=n; x++)
{
for(int j=1; j<=n; j++)
slack[j]=inf;
while(1)
{
memset(visx,0,sizeof(visx));
memset(visy,0,sizeof(visy));
if(dfs(x))
{
break;
}
//这个操作其实就是加边
//方法为:将所有在增广轨中(就是在增广过程中遍历到)的X方点的标号全部减去一个常数d,
//所有在增广轨中的Y方点的标号全部加上一个常数d
int d = inf;
for (int i = 1; i <= n; i ++)
if (!visy[i]&&d > slack[i])
d = slack[i];
for(int i=1; i<=n; i++)
{
if(visx[i]) lx[i]-=d;
if(visy[i]) ly[i]+=d;
else
slack[i]-=d;
}
}
}
int ans=0;
for(int i=1;i<=n;i++)
{
if(girl[i]>-1)
ans+=w[girl[i]][i];
}
return ans;
}
int main()
{
int x,y;
while(scanf("%d",&n)!=EOF)
{
for(int i=1; i<=n; i++)
{
for(int j=1; j<=n; j++)
{
scanf("%d",&w[i][j]);
}
}
printf("%d\n",km());
}
return 0;
}