题意:
给定n*n的矩阵,每个格子有个值s[i][j],现在要求对每行和每列各分配一个值,r[i],c[i]使得对所有的格子都有s[i][j]<=r[i]+c[j]成立,并且sum{r[i]}+sum{c[i]}尽量小。
直接KM啊。。SB题啊。。。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
#define INF 100000000
int n,nx,ny;
int link[510],lx[510],ly[510],slack[510],w[510][510],visx[510],visy[510];
bool dfs(int x)
{
visx[x]=1;
for(int y=1;y<=ny;y++)
{
if(visy[y])
continue;
int t=lx[x]+ly[y]-w[x][y];
if(t==0)
{
visy[y]=1;
if(link[y]==-1||dfs(link[y]))
{
link[y]=x;
return 1;
}
}
else if(slack[y]>t)
slack[y]=t;
}
return 0;
}
void KM()
{
memset(link,-1,sizeof(link));
memset(ly,0,sizeof(ly));
for(int i=1;i<=nx;i++)
{
lx[i]=-INF;
for(int j=1;j<=ny;j++)
lx[i]=max(lx[i],w[i][j]);
}
for(int x=1;x<=nx;x++)
{
for(int i=1;i<=ny;i++)
slack[i]=INF;
while(1)
{
memset(visx,0,sizeof(visx));
memset(visy,0,sizeof(visy));
if(dfs(x))
break;
int d=INF;
for(int i=1;i<=ny;i++)
{
if(!visy[i]&&d>slack[i])
d=slack[i];
}
for(int i=1;i<=nx;i++)
{
if(visx[i])
lx[i]-=d;
}
for(int i=1;i<=ny;i++)
{
if(visy[i])
ly[i]+=d;
else
slack[i]-=d;
}
}
}
int ans=0;
for(int i=1;i<nx;i++)
{
ans+=lx[i];
printf("%d ",lx[i]);
}
ans+=lx[nx];
printf("%d\n",lx[nx]);
for(int i=1;i<ny;i++)
{
ans+=ly[i];
printf("%d ",ly[i]);
}
ans+=ly[ny];
printf("%d\n",ly[ny]);
printf("%d\n",ans);
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
nx=ny=n;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
scanf("%d",&w[i][j]);
}
KM();
}
return 0;
}
原文:http://www.cnblogs.com/water-full/p/4532784.html