KM算法--带权二分匹配

http://acm.hdu.edu.cn/showproblem.php?pid=2255

问题概述:有n个人,n个房子,每个人对每个房子出价都不同,你是村长,你如何分配房子才能获得最高收益?

输入样例:                                对应输出:

5                                                 29

3 4 6 4 9

6 4 5 3 8

7 5 3 4 2

6 3 2 2 5

8 4 5 4 7


http://philoscience.iteye.com/blog/1754498

完备匹配:对于一个二分图,最大匹配数为其点数少的那一边的点数

顶标:每个点的权值,恒满足lx[i]+ly[j]==val[i][j](val[i][j]表示第i个人对第j个房子的出价)

→若由二分图中所有满足lx[i]+ly[j]==val[i][j]的边(i,j)构成的子图有完备匹配,那么这个完备匹配就是二分图的最大权

匹配(可能有多个,但总权值大小相同)

KM算法流程:

①:初始化可行顶标的值,默认lx[i] = val[i][j], ly[j] = 0

②:用匈牙利算法尽可能寻找完备匹配,当然满足lx[i]+ly[j]==val[i][j]才匹配

③:若未找到完备匹配则修改可行顶标的值,例如在满足lx[i]+ly[j]==val[i][j]条件下降低lx[i],增加对应的ly[j]

④:重复②③直到找到相等子图的完备匹配为止


#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n, link[305], lx[305], ly[305], slack[305], visx[305], visy[305], val[305][305];
int Sech(int x);			/*lx、ly为顶标,slack为松弛函数,对应Y集合*/
int main(void)
{
	int i, j, ans, d, k;
	while(scanf("%d", &n)!=EOF)
	{
		n;
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
				scanf("%d", &val[i][j]);
		}
		memset(link, -1, sizeof(link));
		memset(ly, 0, sizeof(ly));
		for(i=1;i<=n;i++)
		{										/* lx初始化为与它关联边中最大的 */
			lx[i] = 0;
			for(j=1;j<=n;j++)
				lx[i] = max(val[i][j], lx[i]);
		}
		for(k=1;k<=n;k++)
		{
			for(i=1;i<=n;i++)
				slack[i] = 10000000;
			while(1)
			{
				memset(visx, 0, sizeof(visx));
				memset(visy, 0, sizeof(visy));
				if(Sech(k))							/*若成功找到增广路,则该点匹配完成*/
					break;								/*若失败,则需要改变一些点的顶标,使得图中可行边的数量增加*/
				d = 10000000;
				for(i=1;i<=n;i++)						/*所有在增广过程中遍历到的X方点的顶标全部减去一个常数d,Y放点的顶标全部加一个d(d尽可能小)*/
				{
					if(visy[i]==0)
						d = min(d, slack[i]);
				}
				for(i=1;i<=n;i++)
				{
					if(visx[i])
						lx[i] -= d;
					if(visy[i])
						ly[i] += d;
					else
						slack[i] -= d;		/*修改顶标后,要把所有不在交错树中的Y顶点的slack值都减去d*/
				}
			}
		}
		ans = 0;
		for(i=1;i<=n;i++)
		{
			//if(link[i]!=-1)
				ans += val[link[i]][i];
		}
		printf("%d\n", ans);
	}
	return 0;
}

int Sech(int x)				/*匈牙利算法*/
{
	int i, temp;
	visx[x] = 1;			/*深搜寻找从x点开始的增广路*/
	for(i=1;i<=n;i++)
	{
		if(visy[i])
			continue;
		temp = lx[x]+ly[i]-val[x][i];
		if(temp==0)
		{
			visy[i] = 1;
			if(link[i]==-1 || Sech(link[i]))
			{
				link[i] = x;
				return 1;
			}
		}
		else								/*如果lx[x]+ly[i]>val[x][i]*/
			slack[i] = min(slack[i], temp);
	}
	return 0;
}



  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值