您的位置:首页 > 其它

二分图最大权匹配--KM算法模板

2016-05-30 16:32 483 查看
KM算法我是看了训练指南学会的,指出上面的一处错误,当时纠结了好久才发现,在349页情况二中第一个S应该是S‘。

模板代码如下:

import java.util.*;

public class Main {
static int n,min;
static int[] ss=new int[105];
static int[] tt=new int[105];
static int[] lx=new int[105];
static int[] ly=new int[105];
static int[] match=new int[105];
static int[][] map=new int[105][105];
static void km()
{
for(int i=1;i<=n;i++)
{
lx[i]=ly[i]=match[i]=0;
for(int j=1;j<=n;j++)
lx[i]=Math.max(lx[i],map[i][j]);
}
for(int i=1;i<=n;i++)
{
while(true)
{
Arrays.fill(ss,0);
Arrays.fill(tt,0);
min=1000000000;
if(dfs(i)) break;
for(int j=1;j<=n;j++)
{
if(ss[j]==1) lx[j]-=min;
if(tt[j]==1) ly[j]+=min;
}
}
}
}
static boolean dfs(int x)
{
ss[x]=1;
for(int i=1;i<=n;i++)
{
if(tt[i]==1) continue;
if(lx[x]+ly[i]!=map[x][i])
min=Math.min(min,lx[x]+ly[i]-map[x][i]);
else
{
tt[i]=1;
if(match[i]==0||dfs(match[i]))
{
match[i]=x;break;
}
}
}
return false;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: