带权值的二分图匹配 KM算法

1，如果二分图不是完全二分图，我们通过添加无用路径（最大匹配中，路径权值为0）和顶点使之成为完全二分图；

2，使用KM算法求解，KM算法核心需要理解feasible vertex labeling和equality subgraph概念，在equality subgraph中寻找最大匹配（采用匈牙利算法），如果最大匹配正好为完全匹配，根据KM理论，这个完全匹配就是带权值的最大匹配；如果在当前的equality subgraph获取的最大匹配不是完全匹配，我们通过KM算法中提供的修改label方法，增加新的y点，得到新的equality subgraph，再继续在新的subgraph寻找最大匹配，如此循环。

代码如下：

#include <cstdio>
#include <memory.h>
#include <algorithm>    // 使用其中的 min 函数
using namespace std;

const int MAX = 1024;

int n; // X 的大小
int weight[MAX][MAX]; // X 到 Y 的映射（权重）
int lx[MAX], ly[MAX]; // 标号
bool sx[MAX], sy[MAX]; // 是否被搜索过
int match[MAX]; // Y(i) 与 X(match [i]) 匹配

// 初始化权重
void init(int size);
// 从 X(u) 寻找增广道路，找到则返回 true
bool path(int u);
// 参数 maxsum 为 true ，返回最大权匹配，否则最小权匹配
int bestmatch(bool maxsum = true);

void init(int size)
{
	// 根据实际情况，添加代码以初始化
	n = size;
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			scanf("%d", &weight[i][j]);
}
/*
 * 和二分图的思路类似，在子图中寻找增广路径
 */
bool path(int u)
{
	sx[u] = true;
	for (int v = 0; v < n; v++)
		if (!sy[v] && lx[u] + ly[v] == weight[u][v])
		{
			sy[v] = true;
			if (match[v] == -1 || path(match[v]))
			{
				match[v] = u;
				return true;
			}
		}
	return false;
}

int bestmatch(bool maxsum)
{
	int i, j;
	if (!maxsum)
	{
		for (i = 0; i < n; i++)
			for (j = 0; j < n; j++)
				weight[i][j] = -weight[i][j];
	}

	// 初始化标号
	for (i = 0; i < n; i++)
	{
		lx[i] = -0x1FFFFFFF;
		ly[i] = 0;
		for (j = 0; j < n; j++)
			if (lx[i] < weight[i][j])
				lx[i] = weight[i][j];
	}

	memset(match, -1, sizeof(match));
	for (int u = 0; u < n; u++)
		while (1)
		{
			memset(sx, 0, sizeof(sx));
			memset(sy, 0, sizeof(sy));
			if (path(u))	//一直寻找增广路径，直到子图中没有增广路径，我们通过修改label来增加新的点，增加的点必为y
				break;

			// 修改标号
			int dx = 0x7FFFFFFF;
			for (i = 0; i < n; i++)
				if (sx[i])
					for (j = 0; j < n; j++)
						if (!sy[j])
							dx = min(lx[i] + ly[j] - weight[i][j], dx);	//找到松弛变量最小的点
			for (i = 0; i < n; i++)
			{
				if (sx[i])
					lx[i] -= dx;
				if (sy[i])
					ly[i] += dx;
			}
		}

	int sum = 0;
	for (i = 0; i < n; i++)
		sum += weight[match[i]][i];

	if (!maxsum)
	{
		sum = -sum;
		for (i = 0; i < n; i++)
			for (j = 0; j < n; j++)
				weight[i][j] = -weight[i][j]; // 如果需要保持 weight [ ] [ ] 原来的值，这里需要将其还原
	}
	return sum;
}

int main()
{
	int n;
	scanf("%d", &n);
	init(n);
	int cost = bestmatch(true);

	printf("%d ", cost);
	for (int i = 0; i < n; i++)
	{
		printf("Y %d -> X %d ", i, match[i]);
	}

	return 0;
}

附带poj2195解法：

/*
 * poj2195.cpp
 *
 *  Created on: 2012-5-9
 *      Author: ict
 */

#include <cstdio>
#include <cstdlib>
#include <string.h>
#include <algorithm>
#include <cmath>
using namespace std;
#define MAX 200
typedef struct GRID
{
	int x;
	int y;
}grid, pgrid;
grid M[MAX], H[MAX];

int n; // X 的大小
int weight[MAX][MAX]; // X 到 Y 的映射（权重）
int lx[MAX], ly[MAX]; // 标号
bool sx[MAX], sy[MAX]; // 是否被搜索过
int match[MAX]; // Y(i) 与 X(match [i]) 匹配

/*
 * 和二分图的思路类似，在子图中寻找增广路径
 */
bool path(int u)
{
	sx[u] = true;
	for (int v = 0; v < n; v++)
		if (!sy[v] && lx[u] + ly[v] == weight[u][v])
		{
			sy[v] = true;
			if (match[v] == -1 || path(match[v]))
			{
				match[v] = u;
				return true;
			}
		}
	return false;
}

int bestmatch(bool maxsum)
{
	int i, j;
	if (!maxsum)
	{
		for (i = 0; i < n; i++)
			for (j = 0; j < n; j++)
				weight[i][j] = -weight[i][j];
	}

	// 初始化标号
	for (i = 0; i < n; i++)
	{
		lx[i] = -0x1FFFFFFF;
		ly[i] = 0;
		for (j = 0; j < n; j++)
			if (lx[i] < weight[i][j])
				lx[i] = weight[i][j];
	}

	memset(match, -1, sizeof(match));
	for (int u = 0; u < n; u++)
		while (1)
		{
			memset(sx, 0, sizeof(sx));
			memset(sy, 0, sizeof(sy));
			if (path(u))	//一直寻找增广路径，直到子图中没有增广路径，我们通过修改label来增加新的点，增加的点必为y
				break;

			// 修改标号
			int dx = 0x7FFFFFFF;
			for (i = 0; i < n; i++)
				if (sx[i])
					for (j = 0; j < n; j++)
						if (!sy[j])
							dx = min(lx[i] + ly[j] - weight[i][j], dx);	//找到松弛变量最小的点
			for (i = 0; i < n; i++)
			{
				if (sx[i])
					lx[i] -= dx;
				if (sy[i])
					ly[i] += dx;
			}
		}

	int sum = 0;
	for (i = 0; i < n; i++)
		sum += weight[match[i]][i];

	if (!maxsum)
	{
		sum = -sum;
		for (i = 0; i < n; i++)
			for (j = 0; j < n; j++)
				weight[i][j] = -weight[i][j]; // 如果需要保持 weight [ ] [ ] 原来的值，这里需要将其还原
	}
	return sum;
}



int main()
{
	int row, col;
	int i, j;
	int ch;
	int mCount, hCount;

	while(1)
	{
		scanf("%d %d", &row, &col);
		getchar();
		if(row == 0 && col == 0)
			break;
		mCount = 0;
		hCount = 0;
		for(i = 0; i < row ; i++)
		{
			for(j = 0; j < col; j++)
			{
				ch = getchar();
				if(ch == 'm')
				{
					M[mCount].x = i;
					M[mCount].y = j;
					mCount++;
				}
				else
					if(ch == 'H')
					{
						H[hCount].x = i;
						H[hCount].y = j;
						hCount++;
					}
			}
			getchar();
		}

		n = mCount;

		for(i = 0 ; i < n; i++)
			for(j = 0; j < n; j++)
			{
				weight[i][j] = abs(M[i].x - H[j].x) + abs(M[i].y - H[j].y);
			}

		printf("%d\n", bestmatch(false));
	}

	return 0;

}