hdu2853Assignment 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 ;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。