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