hdu 2255 - KM算法

题目链接：点击打开链接

KM算法具体了解：点击打开链接

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;
const int mx = 3e2 + 10;
typedef long long ll;
int n,m,w[mx][mx],lx[mx],ly[mx];
int slack[mx],cp[mx];
bool visx[mx],visy[mx];
bool find(int x)
{
	visx[x] = 1;
	for(int i=1;i<=n;i++){
		if(!visy[i]){
			int val = lx[x] + ly[i] - w[x][i];
			if(!val){
				visy[i] = 1;
				if(!cp[i]||find(cp[i])){
					cp[i] = x;
					return 1;
				}
			}else slack[i] = min(slack[i],val);
			//找到没有在增广路径的最小松弛值,使得lx[x]+ly[i]-w[x][i]=0 
		}
	}
	return 0;
}
int KM()
{
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++) slack[j] = inf;
		memset(visx,0,sizeof(visx));
		memset(visy,0,sizeof(visy));
		while(!find(i)){
			int d = inf;
			for(int j=1;j<=n;j++)
			if(!visy[j]) d = min(d,slack[j]);
			for(int j=1;j<=n;j++)//对于增广路径上的左侧减去最小松弛值 
			if(visx[j]) lx[j] -= d;
			for(int j=1;j<=n;j++)
			if(visy[j]) ly[j] += d;//对于增广路径上的右侧点加上d 
			else slack[j] -= d;//相应松弛值也应该减少d 
			memset(visx,0,sizeof(visx));
			memset(visy,0,sizeof(visy));			 
		} 
	}
	int ans = 0;
	for(int i=1;i<=n;i++)
	ans += w[cp[i]][i];
	return ans;
}
int main()
{
	while(~scanf("%d",&n)){
		memset(lx,-inf,sizeof(lx));
		memset(cp,0,sizeof(cp));
		memset(ly,0,sizeof(ly));
		for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)//初始化二分图左侧点最大值为最大边权重 
		scanf("%d",w[i]+j),lx[i] = max(lx[i],w[i][j]);
		printf("%d\n",KM());
	}
	return 0;
}