HDU2853 Assignment KM算法 巧妙变化

Problem Address：http://acm.hdu.edu.cn/showproblem.php?pid=2853

 

【题意】

 

给定一个二分图的所有匹配权值，再给出一个匹配（不是最优），求最优可以增加多少值，同时要求改变的匹配数目最少。

 

【思路】

 

自己的思路错了。后来上网找了，发现这个思路真的很不错。

 

把所有权值扩大到k倍且k>n，至于为什么要扩大以及扩大n倍以上，后面会看到它的作用。

可以肯定的是，扩大后只要得到结果后除以k即可得到原来的答案。

先计算好当前所给匹配的权值总和，之后把这几个匹配的值均加上一。为什么要加一后面可以看到。

就这样进行KM算法的运算，得到答案。

得到最大权值和之后除以k得到的结果就是最优的匹配。

那增加了一的匹配怎么办呢？由于k是大于n的，所以即使选中了所有原来的匹配，增加的数也是小于k的，即会因为整除的性质而被忽略掉。

而总和对k取模，得到的结果就是没有改变的匹配数目。也就是上面因为除法而被忽略掉了的那些数。用结点数减去这个值就是要求的第一个结果。

由于原来的匹配值均增加了一，即相比于其他本来权值相等的匹配有更大的权值，有更大的竞争性，在本来值相等的情况下会优先被选中。

再者，如果原来两个匹配值相差1，那么增加1就会影响结果，但是由于所有匹配都扩大了k倍，那么扩大后就是相差k，再增加1时也就不会有所影响。

至此，答案就出来了。

 

【代码】

 

//名称：KM算法（邻接矩阵）
//用途：用于计算二分图最大权值匹配
//复杂度：O(n*n*n)
//方法：由KM()传入矩阵行n列m及权值矩阵w，返回最大权值匹配
//      左端点为1-n，右端点为1-m
//应用：求最小权值匹配时将权值转化为负数，得到结果再取反即可

#include <iostream>
using namespace std;

#define min(a,b) ((a)<=(b)?(a):(b))

const int maxn = 50;//二分图左端点个数
const int maxm = 50;//二分图右端点个数
const int inf = (1<<30);//定义最大值
  
int w[maxn+5][maxm+5]; //权值邻接矩阵，初始化为-inf，之后填入权值。若不可达则为-inf
int lx[maxn+5], ly[maxm+5]; 
int linky[maxm+5];//存储右端点对应的左端点匹配，-1表示无匹配
bool visx[maxn+5], visy[maxm+5];  
int lack;

int v[maxn+5];
  
bool find(int v, int m)  
{  
    int i, t;  
    visx[v] = true;  
    for (i=1; i<=m; i++)  
    {  
        if (w[v][i]==-inf || visy[i]) continue;  
        t = lx[v] + ly[i] - w[v][i];  
        if (t==0)  
        {  
            visy[i] = true;  
            if (linky[i]==-1 || find(linky[i], m))  
            {  
                linky[i] = v;  
                return true;  
            }  
        }  
        else lack = min(lack, t);  
    }  
    return false;  
}  
  
int KM(int n, int m)
{  
    int i, j;  
    for (i=0; i<=m; i++)  
    {  
        ly[i] = 0;  
        linky[i] = -1;  
    }  
    for (i=1; i<=n; i++)  
    {  
        lx[i] =-inf;  
        for (j=1; j<=m; j++)  
        {  
            if (w[i][j]>lx[i])  
                lx[i] = w[i][j];  
        }  
    }  
    for (i=1; i<=n; i++)  
    {  
        while(true)  
        {  
            for (j=0; j<=n; j++) visx[j] = false;  
            for (j=0; j<=m; j++) visy[j] = false;  
            lack = inf;  
            if (find(i, m)) break;  
	//  if (lack==inf) return -1;  //如果不存在完美匹配，则返回-1
            for (j=1; j<=n; j++) 
            {  
                if (visx[j]) lx[j] -= lack;  
            }  
            for (j=1; j<=m; j++)  
            {  
                if (visy[j]) ly[j] += lack;  
            }  
        }  
    }  
    int ans = 0;  
    for (i=1; i<=m; i++)  
    {  
        if (linky[i]>-1)  
        {  
            ans += w[linky[i]][i]; 
        }  
    }  
    return ans;  //若为最小权值，则返回其相反数
}

int main()
{
	int n, m;
	int i, j;
	int t, x;
	while(scanf("%d %d", &n, &m)!=EOF)
	{
		for (i=1; i<=n; i++)
		{
			for (j=1; j<=m; j++)
			{
				scanf("%d", &w[i][j]);
				w[i][j] *= (n+1);
			}
		}
		t = 0;
		for (i=1; i<=n; i++)
		{
			scanf("%d", &x);
			t -= w[i][x];
			w[i][x]++;
		}
		t += KM(n, m);
		printf("%d %d\n", n-t%(n+1), t/(n+1));
	}
	return 0;
}