KM算法模板
来源:互联网 发布:webassembly与java 编辑:程序博客网 时间:2024/05/17 10:28
View Code
#define MIN(a,b) ((a)<(b)?(a):(b))
#define INF 999999
#define MAX 5110
int n,match[MAX];
bool sx[MAX],sy[MAX];
int lx[MAX],ly[MAX],map[MAX][MAX];
bool path(int u)
{
sx[u]=true;
for(int v=0;v<n;v++)
if(!sy[v]&&lx[u]+ly[v]==map[u][v])
{
sy[v]=true;
if(match[v]==-1||path(match[v]))
{
match[v]=u;
return true;
}
}
return false;
}
int KM(bool truth)//可以不用更改地处理最小或最大权匹配
{
int i,j;
if(!truth)
{
for(i=0;i<n;i++)
for(j=0;j<n;j++)
map[i][j]=-map[i][j];
}
for(i=0;i<n;i++)
{
lx[i]=-INF;
ly[i]=0;
for(j=0;j<n;j++)
if(lx[i]<map[i][j])
lx[i]=map[i][j];
}
memset(match,-1,sizeof(match));
for(int u=0;u<n;u++)
while(1)
{
memset(sx,0,sizeof(sx));
memset(sy,0,sizeof(sy));
if(path(u)) break;
int dmin=INF;
for(i=0;i<n;i++)
if(sx[i])
for(j=0;j<n;j++)
if(!sy[j])
dmin=MIN(lx[i]+ly[j]-map[i][j],dmin);
for(i=0;i<n;i++)
{
if(sx[i])
lx[i]-=dmin;
if(sy[i])
ly[i]+=dmin;
}
}
int sum=0;
for(j=0;j<n;j++)
sum+=map[match[j]][j];
if(!truth)
{
sum=-sum;
for(i=0;i<n;i++)
for(j=0;j<n;j++)
map[i][j]=-map[i][j];
}
return sum;
}
void Map_Init(int m,int n)
{
int x,y,w;
int i,j;
for(i=0;i<n;i++)
for(j=0;j<n;j++)
map[i][j]=INF;
for(i=0;i<m;i++)
{
scanf("%d%d%d",&x,&y,&w);
if(map[x-1][y-1]>w)
map[x-1][y-1]=w;
}
}
编辑器加载中...
- KM算法模板
- KM算法 详解+模板
- KM算法详解+模板
- KM算法 详解+模板
- KM算法 详解+模板
- KM算法模板
- KM算法模板
- KM算法 详解+模板
- KM算法 详解+模板
- KM算法 详解+模板
- KM算法模板
- KM算法模板 hdu2255
- KM算法 详解+模板
- KM算法详解,模板
- KM算法模板
- KM算法模板
- KM算法模板(HDU_2255)
- KM算法模板
- poj 2002 哈希函数的设计
- hdu 1496 哈希
- poj 2411 状态压缩DP 铺砖块
- 最小费用最大流模板 poj 2135
- hdu 1853 费用流
- KM算法模板
- codeforces 111 160E 线段树
- hdu 2836 树状数组 + DP
- URAL 1846 线段树 最大公约数
- poj 旋转卡壳三题 poj 2187 poj 2079
- hdu 1154 poj 2462 Cutting a polygon 计算几何
- 计算几何模板 更新中
- 半平面交入门 poj 3335 poj 1474 poj 1279 半平面交模板
- poj 1514 Metal Cutting 半平面交的平面切割