【KM】

#include<cstdio>
#include<cstring>
using namespace std;

const int N = 50, INF = 1e9;

int nx, ny;
bool adj[N][N];
int v[N][N];

bool vx[N], vy[N];
int mx[N], my[N];
int lx[N], ly[N], slack[N];

void init(){
	scanf("%d%d", &nx, &ny);
	for(int i=0; i<nx; i++) for(int j=0; j<ny; j++) scanf("%d", &v[i][j]), adj[i][j] = (v[i][j]>0);
}

bool match(int x){
	
	vx[x] = 1;
	for(int y=0; y<ny; y++) if(adj[x][y] && !vy[y]){
		
		int t = lx[x] + ly[y] - v[x][y];
		if(t==0){
			vy[y] = 1;
			if(my[y]==-1 || match(my[y])){
				mx[x] = y; my[y] = x; return 1;
			}
		}
		else if(t<slack[y]) slack[y]=t;
	}
	return 0;
	
}

int KM(){
	
	memset(lx, 0, sizeof(lx));
	memset(ly, 0, sizeof(ly));
	for(int i=0; i<nx; i++) for(int j=0; j<ny; j++) if(adj[i][j] && v[i][j]>lx[i]) lx[i]=v[i][j];
	
	memset(mx, -1, sizeof(mx));
	memset(my, -1, sizeof(my));
	
	for(int x=0; x<nx; x++){
		
		for(int y=0; y<ny; y++) slack[y] = INF;
		
		for(;;){
			
			memset(vx, 0, sizeof(vx));
			memset(vy, 0, sizeof(vy));
			if(match(x)) break;
			
			int aug = INF;
			for(int y=0; y<ny; y++) if(!vy[y] && slack[y]<aug) aug = slack[y];
			for(int i=0; i<nx; i++) if(vx[i]) lx[i] -= aug;
			for(int y=0; y<ny; y++) if(vy[y]) ly[y] += aug; else slack[y]-=aug;
			
		}
		
	}
	
	int ret = 0;
	for(int x=0; x<nx; x++) ret += lx[x];
	for(int y=0; y<ny; y++) ret += ly[y];
	return ret;
	
}

int main(){
	
	init();
	printf("%d\n", KM());
	
	return 0;
}

【1】演算法筆記,Matching

这里把整个匹配都讲的十分详细了。但里面讲KM这一块的标程,明白其中心思路就行,实现不太优美。

【2】求最大权二分匹配的KM算法

标程要看这里的。(上面都是借鉴的)

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值