感谢 http://www.cnblogs.com/vongang/archive/2012/04/28/2475731.html
这篇blog里提供了3个链接……基本上很明白地把KM算法是啥讲清楚了
然而n^4的KM好像并没有什么卵用啊……所以不得不学n^3的
我看了一下各种,大部分blog里写的声称是n^3的KM,其实貌似都是n^4的(包括上面的链接以及上面链接里提供的链接)
这是因为他们有个共同点
他们虽然用slack数的优化组避免了暴力枚举d所消耗的时间,但由于一次增广是n^2的,所以拖慢了复杂度
那么怎么解决这个问题呢?
尛焱轟告诉我们,用bfs增广的KM是n^3的,用dfs增广的KM是n^4的
尛焱轟还告诉我们,可以去UOJ上拉个板子,都是n^3的
于是窝就拉了个策爷的板子来看(然后改了几个变量名,背下来,就学完了……)
为什么dfs会成为算法时间复杂度减小的瓶颈呢?
我们发现,每更新顶标,就要重新从当前点开始dfs找一遍增广路,有很多冗余的操作
实际上,更新完顶标之后,交错树只会增加新的点
那么窝萌不妨用bfs来增广,每次修改完顶标,把没访问到的右侧点的slack值也相应地减去d,那么slack值为0就相当于多了一条可行边,就相当于能够访问到新的节点,也就可以继续找增广路了
这样再把新的点加进队列,就避免了dfs增广的版本中的冗余操作
这样就发挥了slack这一优化的优势,复杂度自然降到O(n^3)
然后窝来帖一下窝的代码(uoj#80 二分图最大权匹配)
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#define ll long long
#define N 405
#define INF (1LL<<60) using namespace std;
inline int read(){
int ret=0;char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while ('0'<=ch&&ch<='9'){
ret=ret*10-48+ch;
ch=getchar();
}
return ret;
} int n,fx[N],fy[N],prev[N];
ll g[N][N],A[N],B[N],slk[N];
bool visx[N],visy[N];
int q[N],qh,qt; void aug(int v){
if (!v) return;
fy[v]=prev[v];
aug(fx[prev[v]]);
fx[fy[v]]=v;
} void bfs_KM(int _s){
memset(visx,0,sizeof(visx));
memset(visy,0,sizeof(visy));
memset(slk,127,sizeof(slk));
qh=qt=0;
q[++qt]=_s;
for (;;){
while (qh<qt){
int u=q[++qh];
visx[u]=1;
for (int v=1;v<=n;++v)if (!visy[v]){
if (A[u]+B[v]==g[u][v]){
visy[v]=1;
prev[v]=u;
if (!fy[v]){aug(v);return;}
q[++qt]=fy[v];
continue;
}
if (slk[v]>A[u]+B[v]-g[u][v]){
slk[v]=A[u]+B[v]-g[u][v];
prev[v]=u;
}
}
}
ll d=INF;
for (int i=1;i<=n;++i)
if (!visy[i]) d=min(d,slk[i]);
for (int i=1;i<=n;++i){
if (visx[i]) A[i]-=d;
if (visy[i]) B[i]+=d;
else slk[i]-=d;
}
for (int v=1;v<=n;++v)if (!visy[v]&&!slk[v]){
visy[v]=1;
if (!fy[v]){aug(v);return;}
q[++qt]=fy[v];
}
}
} ll KM(){
for (int i=1;i<=n;++i){
A[i]=-INF;B[i]=0;
for (int j=1;j<=n;++j)
A[i]=max(A[i],g[i][j]);
}
memset(fx,0,sizeof(fx));
memset(fy,0,sizeof(fy));
for (int i=1;i<=n;++i) bfs_KM(i);
ll ret=0;
for (int i=1;i<=n;++i) ret+=A[i]+B[i];
return ret;
} bool e0[N][N];
int main(){
int nl=read(),nr=read();
memset(g,0,sizeof(g));
memset(e0,0,sizeof(e0));
for (int m0=read();m0;--m0){
int u=read(),v=read();
g[u][v]=read();
e0[u][v]=1;
}
n=max(nl,nr);
ll ans=KM();
printf("%lld\n",ans);
for (int i=1;i<=nl;++i)
printf("%d ",e0[i][fx[i]]*fx[i]);
puts("");
return 0;
}
感谢尛焱轟神犇的指点
感谢jcvb神犇的代码
感谢上面的那篇blog以及那篇blog里的链接
更新一下,窝在丘比特的烦恼(KM模板题)里把KM封装了一下,方便大(我)家(拖)学(板)习(子)
顺便提一下此题的几个坑爹的地方:1坐标可能为负,2姓名无视大小写,3必须连n对情侣(也就是说不能连的边权必须赋为-INF)
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <map>
#include <string>
#define N 32
#define INF (1e9) using namespace std;
inline int read(){
int ret=0;char ch=getchar();
bool flag=0;
while (ch<'0'||ch>'9'){
flag=ch=='-';
ch=getchar();
}
while ('0'<=ch&&ch<='9'){
ret=ret*10-48+ch;
ch=getchar();
}
return flag?-ret:ret;
} struct KM{
int n;
int g[N][N],slk[N],A[N],B[N];
int prev[N],fx[N],fy[N];
bool visx[N],visy[N];
int q[N],qh,qt;
void clear(int _n){
n=_n;memset(g,0,sizeof(g));
}
void AddEdge(int u,int v,int w){
g[u][v]=w;
}
void aug(int v){
if (!v) return;
fy[v]=prev[v];
aug(fx[fy[v]]);
fx[fy[v]]=v;
}
void bfs(int _s){
memset(visx,0,sizeof(visx));
memset(visy,0,sizeof(visy));
memset(slk,127,sizeof(slk));
qh=qt=0;q[++qt]=_s;
for (;;){
while (qh<qt){
int u=q[++qh];
visx[u]=1;
for (int v=1;v<=n;++v)if (!visy[v]){
if (A[u]+B[v]==g[u][v]){
visy[v]=1;
prev[v]=u;
if (!fy[v]){aug(v);return;}
q[++qt]=fy[v];
}
else if (slk[v]>A[u]+B[v]-g[u][v]){
slk[v]=A[u]+B[v]-g[u][v];
prev[v]=u;
}
}
}
int d=INF;
for (int i=1;i<=n;++i)if (!visy[i])d=min(d,slk[i]);
for (int i=1;i<=n;++i){
if (visx[i]) A[i]-=d;
if (visy[i]) B[i]+=d;
else slk[i]-=d;
}
for (int v=1;v<=n;++v)
if (!visy[v]&&!slk[v]){
visy[v]=1;
if (!fy[v]){aug(v);return;}
q[++qt]=fy[v];
}
}
}
int solve(){
memset(A,128,sizeof(A));
memset(B,0,sizeof(B));
memset(fx,0,sizeof(fx));
memset(fy,0,sizeof(fy));
for (int i=1;i<=n;++i)
for (int j=1;j<=n;++j) A[i]=max(A[i],g[i][j]);
for (int i=1;i<=n;++i) bfs(i);
int ret=0;
for (int i=1;i<=n;++i) ret+=A[i]+B[i];
return ret;
}
} km; int n;
map<string,int> id;
int x[N*2],y[N*2],lmt;
void Upper(string &s){
int l=s.length();
for (int i=0;i<l;++i)if (s[i]>'Z') s[i]-=32;
} int main(){
string tmp;
lmt=read();n=read();
for (int i=1;i<=2*n;++i){
x[i]=read();y[i]=read();
cin>>tmp;Upper(tmp);id[tmp]=i;
}
km.clear(n);
for (int i=1;i<=n;++i)
for (int j=1;j<=n;++j)
km.AddEdge(i,j,1);
for (cin>>tmp;tmp!="End";cin>>tmp){
Upper(tmp);
int u=id[tmp],v;
cin>>tmp;Upper(tmp);v=id[tmp];
if (u>v) swap(u,v);
km.AddEdge(u,v-n,read());
}
for (int i=1;i<=n;++i)
for (int j=n+1;j<=2*n;++j){
bool found=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])>lmt*lmt;
for (int k=1;k<=2*n&&!found;++k){
int A=y[k]-y[j],B=x[k]-x[j],C=y[k]-y[i],D=x[k]-x[i];
found=A*D==B*C&&(A*C<0||B*D<0);
}
if (found) km.AddEdge(i,j-n,-1e7);
}
printf("%d\n",km.solve());
return 0;
}