HDU-2255 奔小康赚大钱

二分图最大权匹配模板题,用了KM算法,也可用最小费用最大流

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=300+10;
const int INF=0x7f7f7f7f;
int nx,ny;
int g[N][N];
int linker[N],lx[N],ly[N];
int slack[N];
bool visx[N],visy[N];
bool dfs(int x)
{
	visx[x]=true;
	for(int y=0;y<ny;y++)
	{
		if(visy[y]) continue;
		int tmp=lx[x]+ly[y]-g[x][y];
		if(tmp==0)
		{
			visy[y]=true;
			if(linker[y]==-1||dfs(linker[y]))
			{
				linker[y]=x;
				return true;
			}
		}
		else if(slack[y]>tmp)
			slack[y]=tmp;
	}
	return false;
}
int KM()
{
	memset(linker,-1,sizeof(linker));
	memset(ly,0,sizeof(ly));
	for(int i=0;i<nx;i++)
	{
		lx[i]=-INF;
		for(int j=0;j<ny;j++)
			if(g[i][j]>lx[i])
				lx[i]=g[i][j];
	}
	for(int x=0;x<nx;x++)
	{
		for(int i=0;i<ny;i++)
			slack[i]=INF;
		while(true)
		{
			memset(visx,false,sizeof(visx));
			memset(visy,false,sizeof(visy));
			if(dfs(x)) break;
			int d=INF;
			for(int i=0;i<ny;i++)
				if(!visy[i]&&d>slack[i])
					d=slack[i];
			for(int i=0;i<nx;i++)
				if(visx[i]) lx[i]-=d;
			for(int i=0;i<ny;i++)
			{
				if(visy[i]) ly[i]+=d;
				else slack[i]-=d;
			}
		}
	}
	int res=0;
	for(int i=0;i<ny;i++)
		if(linker[i]!=-1)
			res+=g[linker[i]][i];
	return res;
}
int main()
{
	int n;
	while(~scanf("%d",&n))
	{
		nx=ny=n;
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				scanf("%d",&g[i][j]);
		printf("%d\n",KM());
	}
    return 0;
}

猜你喜欢

转载自blog.csdn.net/wl16wzl/article/details/83241691