标签:km算法
//n个人,匹配m个任务,每个人完成每个任务的效率不同
//刚开始已经有了匹配方案,现在重新设计匹配方案,使得效率最高,且尽量保存
//原来的匹配方案,
//将所有权值*1000, 然后对于原来匹配的边的权值+1,用KM算法求出最大带权匹配ans
//那么最终的最大效率为ans/1000 , 保留原来的匹配边的个数为ans%mod
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std ;
const int maxn = 60 ;
const int mod = 1000 ;
const int inf = 0x3f3f3f3f ;
int match[maxn] , lx[maxn] , ly[maxn] , slack[maxn] ;
int w[maxn][maxn] , visx[maxn] , visy[maxn] ;
int n , m ;
bool find(int x)
{
visx[x] = 1 ;
for(int i = 1;i <= m;i++)
{
if(visy[i])continue ;
int tmp = lx[x] + ly[i] - w[x][i] ;
if(tmp == 0)
{
visy[i] = 1 ;
if(match[i] == -1 || find(match[i]))
{
match[i] = x ;
return true ;
}
}
else slack[i] = min(slack[i] , tmp) ;
}
return false ;
}
int KM()
{
memset(match , -1 ,sizeof(match)) ;
memset(ly , 0 , sizeof(ly)) ;
for(int i = 1;i <= n;i++)
{
lx[i] = -inf ;
for(int j = 1;j <= m;j++)
lx[i] = max(lx[i] , w[i][j]) ;
}
for(int i = 1;i <= n;i++)
{
for(int j = 1;j <= m;j++)
slack[j] = inf ;
while(1)
{
memset(visx , 0 , sizeof(visx)) ;
memset(visy , 0 , sizeof(visy)) ;
if(find(i))break;
int d = inf ;
for(int j = 1;j <= m;j++)
if(!visy[j])
d = min(d , slack[j]) ;
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 ;
else
slack[j] -= d ;
}
}
int ans = 0 ;
for(int i = 1;i <= m;i++)
if(match[i] != -1)
ans += w[match[i]][i] ;
return ans ;
}
int main()
{
//freopen("in.txt" , "r" , stdin) ;
while(~scanf("%d%d" ,&n , &m))
{
for(int i = 1;i <= n;i++)
for(int j = 1;j <= m;j++)
{
scanf("%d" , &w[i][j]) ;
w[i][j] *= mod ;
}
int sum = 0 ;
for(int i = 1;i <= n;i++)
{
int a ;
scanf("%d" , &a) ;
sum += w[i][a]/mod ;
w[i][a]++ ;
}
int ans = KM() ;
printf("%d %d\n" , n - ans%mod , ans/mod - sum) ;
}
return 0 ;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
标签:km算法
原文地址:http://blog.csdn.net/cq_pf/article/details/47343317