KM算法的真O(n^3)写法

最近研究了下KM算法求二分图的最大权匹配,网上一堆人讲slack数组可以优化KM算法到O(n3),仔细研究了下发现他们的写法和直接找min值,复杂度本质上是一样的,感觉有点画蛇添足。
看了一篇大佬的文章,找到了这个真O(n3)的写法,它通过不重复构造新的增广路径将KM算法优化到了O(n3),这里我自己实现了一遍。以下是实现代码。
大佬的链接

#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 10000

int nx,ny;
bool visx[maxn],visy[maxn];
int fa[2*maxn];
int slack[maxn];
int match[maxn];
int cx[maxn],cy[maxn];
int G[maxn][maxn];

int dfs(int x)
{
	visx[x]=1;
	for(int y=0;y<=ny-1;y++)
	{
		if(!visy[y])
		{
			int tmp=cx[x]+cy[y]-G[x][y];
			if(tmp==0)
			{
				visy[y]=1;
				fa[y+nx]=x;
				if(match[y]==-1)
				{
					return y+nx;
				}
				fa[match[y]]=y+nx;
				int res=dfs(match[y]);
				if(res>=0)
				{
					return res;
				}
				
			}
			else
			{
				slack[x]=min(slack[x],tmp);
			}
		}
	
	}
	return -1;
}
void km()
{
	memset(match,-1,sizeof(match));
	for(int i=0;i<=nx-1;i++)
	{
		for(int j=0;j<=ny-1;j++)
		{
			cx[i]=max(cx[i],G[i][j]);
		}
	}
	for(int x=0;x<=nx-1;x++)
	{
		memset(visx,0,sizeof(visx));
		memset(visy,0,sizeof(visy));
		memset(slack,INF,sizeof(slack));
		memset(fa,-1,sizeof(fa));
		bool first=true;
		int end=-1;
		while(1)
		{
			if(first)
			{
				first=false;
				end=dfs(x);
			}
			else
			{
				for(int i=0;i<=nx-1;i++)
				{
					if(slack[i]==0)
					{
						slack[i]=INF;
						end=dfs(i);
						if(end>=0)
							break;
					}	
				}	
			}
			if(end>=0)
			{
				int p=end;
				while(p!=-1)
				{
					match[p-nx]=fa[p];
					p=fa[fa[p]];
				}
				break;
			}
			else
			{
				int d=INF;
				for(int i=0;i<=nx-1;i++)
				{
					d=min(d,slack[i]);
				}
				for(int i=0;i<=nx-1;i++)
				{
					if(visx[i])
					{
						cx[i]-=d;
						slack[i]-=d;
					}
				}
				for(int i=0;i<=ny-1;i++)
				{
					if(visy[i])
					{
						cy[i]+=d;
					}
				}
				
			}
			
		}
	}
	
 } 
int main()
{
	int n;
	scanf("%d",&n);
	nx=ny=n;
	for(int i=0;i<=nx-1;i++)
	{
		for(int j=0;j<=ny-1;j++)
		{
			int tmp;
			scanf("%d",&tmp);
			G[i][j]=tmp;
		}
	}
	km();
	for(int i=0;i<=ny-1;i++)
	{
		printf("%d ",match[i]);
	}
	printf("\n");
	int sum=0;
	for(int i=0;i<=ny-1;i++)
	{
		sum+=G[match[i]][i];
	}
	printf("%d",sum);
	return 0;
 } 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值