hdu1533Going Home KM算法

标签：

//给一个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;
    }
}

hdu1533Going Home KM算法

标签：

原文地址：http://blog.csdn.net/cq_pf/article/details/47724431