hdu1533Going Home KM算法

来源：互联网发布：mysql字段名大小写编辑：程序博客网时间：2024/06/13 21:26
//给一个n*m的图，//m表示人，h表示房子//问所有人走回家的最小步数//每个人只能进一间房//很明显的最大带权匹配//每个人到每每间房的距离即为权值//由于是求最小，只要改为负权就行#include<cstdio>#include<cstring>#include<iostream>#include<queue>#include<cmath>using namespace std ;const int maxn = 110 ;const int inf  = 0x3f3f3f3f ;int map[maxn][maxn] ;int lx[maxn] ,ly[maxn] ;int slack[maxn] , match[maxn] ;int visx[maxn] , visy[maxn] ;char str[maxn][maxn] ;int mx[maxn] , my[maxn] ;int hx[maxn] , hy[maxn] ;int n , m ;bool find(int x){    visx[x] = 1 ;    for(int i = 1;i <= n;i++)    {        if(visy[i])continue ;        int tmp = lx[x] + ly[i] - map[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 <= n;j++)         lx[i] = max(lx[i] , map[i][j]) ;   }   for(int i = 1;i <= n;i++)   {       for(int j = 1;j <= n;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 <= n;j++)             if(!visy[j])               d = min(slack[j] , d) ;           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 ;               else                 slack[j] -= d ;       }   }   int ans = 0 ;   for(int j = 1;j <= n;j++)   ans += map[match[j]][j];   return ans ;}int main(){   // freopen("in.txt" , "r" , stdin) ;    while(scanf("%d%d" , &n , &m) && (n+m))    {        int len_m = 0 , len_h = 0 ;        for(int i = 1;i <= n;i++)        {            scanf("%s" , &str[i][1]) ;            for(int j = 1;j <= m;j++)            if(str[i][j] == 'm')            {                mx[++len_m] = i ;                my[len_m] = j ;            }            else if(str[i][j] == 'H')            {                hx[++len_h] = i ;                hy[len_h] = j ;            }        }        memset(map , 0 ,sizeof(map)) ;        for(int i = 1;i <= len_m;i++)            for(int j = 1;j <= len_h;j++)            map[i][j] = (int)(-abs((double)(mx[i]-hx[j])) - abs((double)(my[i]-hy[j]))) ;        n = len_m ;        int ans = KM() ;        cout<<-ans<<endl;    }}
0 0