还是没看懂一般图都是最大匹配问题。。怪我太笨了哎~ 先来个看明白了的KM算法——寻找带权二分图的最佳匹配方法
一般对KM算法的描述,基本上可以概括成以下几个步骤:
(1) 初始化可行标杆KM算法的最大特点在于利用标杆和权重来生成一个二分子图,在该二分子图上面找最大匹配,而且,当些仅当找到完备匹配,才能得到最佳匹配。标杆和权重的作用在于限制新边的加入,使得加入的新边总是能为子图添加匹配数,同时又令权重和得到最大的提高。
代码模板
#include
#include
#include
#include
using namespace std;
int const MAX = 1000;
int const inf = 0x3fffffff;
int w[MAX][MAX];
int link[MAX];//代表当前与Y集合中配对的X集合中的点
int visx[MAX], visy[MAX];
int lx[MAX], ly[MAX];
int n, m;//代表X和Y中元素的个数
int can(int t)
{
visx[t] = 1;
for(int i = 1; i <= m; i++){
if(!visy[i] && lx[t] + ly[i] == w[t][i]){//这里“lx[t]+ly[i]==w[t][i]”决定了这是在相等子图中找增广路的前提,非常重要
visy[i] = 1;
if(link[i] == -1 || can(link[i])){
link[i] = t;
return 1;
}
}
}
return 0;
}
int km()
{
int sum = 0;
memset(ly, 0, sizeof(ly));
for(int i = 1; i <= n; i++){//把各个lx的值都设为当前w[i][j]的最大值
lx[i] = -inf;
for(int j = 1; j <= n; j++){
if(lx[i] < w[i][j])
lx[i] = w[i][j];
}
}
memset(link, -1, sizeof(link));
for(int i = 1; i <= n; i++){
while(1){
memset(visx, 0, sizeof(visx));
memset(visy, 0, sizeof(visy));
if(can(i))//如果它能够形成一条增广路径,那么就break
break;
int d = inf;//否则,后面应该加入新的边,这里应该先计算d值
for(int j = 1; j <= n; j++)//对于搜索过的路径上的XY点,设该路径上的X顶点集为S,Y顶点集为T,对所有在S中的点xi及不在T中的点yj
if(visx[j])
for(int k = 1; k <= m; k++)
if(!visy[k])
d = min(d, lx[j] + ly[k] - w[j][k]);
if(d == inf)
return -1;//找不到可以加入的边,返回失败(即找不到完美匹配)
for (int j = 1; j <= n; j++)
if (visx[j])
lx[j] -= d;
for(int j = 1; j <= m; j++)
if(visy[j])
ly[j] += d;
}
}
for(int i = 1; i <= m; i++)
if(link[i] > -1)
sum += w[link[i]][i];
return sum;
}
input | output |
---|---|
4 62 41 86 94 73 58 11 12 69 93 89 88 81 40 69 13 |
650 |
问题分析:
有n个垃圾桶,每个垃圾桶内有n种垃圾,现在是要你把垃圾分类,到每一个垃圾桶最后都只能装一种垃圾,从一个垃圾桶里把垃圾移到另一个垃圾桶会有所消耗,且都是单位消耗。。
那么,要总消耗最少,就要原来每个垃圾桶各自保留的垃圾总和要最多,这么看来,问题就可以转化为带权二分图的最佳匹配问题。
求出这个最佳匹配 k 最后 ans = s(所有垃圾总和)-k
代码
#include
#include
#include
using namespace std;
const int MAX = 200;
const int inf = 0x3f3f3f3f;
int w[MAX][MAX];
int lx[MAX], ly[MAX];
int visx[MAX], visy[MAX];
int link[MAX];
int n;
int can(int t)
{
visx[t] = 1;
for(int i = 1; i <= n; i++){
if(!visy[i] && lx[t] + ly[i] == w[t][i]){
visy[i] = 1;
if(link[i] == -1 || can(link[i])){
link[i] = t;
return 1;
}
}
}
return 0;
}
int km()
{
int d;
int sum = 0;
memset(ly, 0, sizeof(ly));
for(int i = 1; i <= n; i++){
lx[i] = -inf;
for(int j = 1; j <= n; j++)
lx[i] = max(lx[i], w[i][j]);
}
memset(link, -1, sizeof(link));
for(int i = 1; i <= n; i++){
while(1){
memset(visx, 0, sizeof(visx));
memset(visy, 0, sizeof(visy));
if(can(i)) break;
for(int j = 1; j <= n; j++)
if(visx[j])
for(int k = 1; k <= n; k++)
if(!visy[k])
d = min(d, lx[j] + ly[k] - w[j][k]);
if(d == inf)
return -1;
for(int j = 1; j <= n; j++)
if(visx[j])
lx[j] -= d;
for(int j = 1; j <= n; j++)
if(visy[j])
ly[j] += d;
}
}
for(int i = 1; i <= n; i++)
if(link[i] > -1)
sum += w[link[i]][i];
return sum;
}
int main()
{
int ans = 0;
scanf("%d", &n);
for(int i = 1; i <= n; i++)
for(int j = 1; j <= n; j++){
scanf("%d", &w[i][j]);
ans += w[i][j];
}
ans -= km();
printf("%d\n", ans);
return 0;
}