HDU 1533

http://acm.hdu.edu.cn/showproblem.php?pid=1533

人和房子数量相同，每个人进房子，费用是人到房子的曼哈顿距离，求最小费用

可用最小费用最大流求解，建立虚拟的源点和汇点即可

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std ;
int ABS(int x)
{
    return x>0?x:-x ;
}
struct point{
    int x,y ;
}mm[100005],hh[100005] ;
const int INF=0xfffffff ;
struct node{
    int s,t,cap,cost,nxt ;
}e[200005] ;
int sumflow ;
int n,m,cnt,head[1005],vis[1005],dis[1005],pre[1005] ;
char M[1005][1005] ;
void add(int s,int t,int cap,int cost)
{
    e[cnt].s=s ;e[cnt].t=t ;e[cnt].cap=cap ;e[cnt].cost=cost ;e[cnt].nxt=head[s] ;head[s]=cnt++ ;
    e[cnt].s=t ;e[cnt].t=s ;e[cnt].cap=0 ;e[cnt].cost=-cost ;e[cnt].nxt=head[t] ;head[t]=cnt++ ;
}
int spfa(int s,int t,int N)
{
    for(int i=0 ;i<=N ;i++)
        dis[i]=INF ;
    dis[s]=0 ;
    memset(vis,0,sizeof(vis)) ;
    memset(pre,-1,sizeof(pre)) ;
    vis[s]=1 ;
    queue <int> q ;
    q.push(s) ;
    while(!q.empty())
    {
        int u=q.front() ;
        q.pop() ;
        vis[u]=0 ;
        for(int i=head[u] ;i!=-1 ;i=e[i].nxt)
        {
            int tt=e[i].t ;
            if(e[i].cap && dis[tt]>dis[u]+e[i].cost)
            {
                dis[tt]=dis[u]+e[i].cost ;
                pre[tt]=i ;
                if(!vis[tt])
                {
                    vis[tt]=1 ;
                    q.push(tt) ;
                }
            }
        }
    }
    if(dis[t]==INF)return 0 ;
    return 1 ;
}
int MCMF(int s,int t,int N)
{
    int flow,minflow,mincost ;
    mincost=flow=0 ;
    while(spfa(s,t,N))                                                
    {
        minflow=INF ;
        for(int i=pre[t] ;i!=-1 ;i=pre[e[i].s])
            minflow=min(minflow,e[i].cap) ;
        flow+=minflow ;
        for(int i=pre[t] ;i!=-1 ;i=pre[e[i].s])
        {
            e[i].cap-=minflow ;
            e[i^1].cap+=minflow ;
        }
        mincost+=dis[t]*minflow ;
    }
    sumflow=flow ;//最大流 
    return mincost ;
}
int main()
{
    while(scanf("%d%d",&n,&m))
    {
        if(!n && !m)break ;
        cnt=0 ;
        memset(head,-1,sizeof(head)) ;
        for(int i=0 ;i<n ;i++)
            scanf("%s",M[i]) ;
        int cm,ch ;
        cm=ch=0 ;
        for(int i=0 ;i<n ;i++)
        {
            for(int j=0 ;j<m ;j++)
            {
                if(M[i][j]=='m')
                {
                    mm[cm].x=i ;
                    mm[cm++].y=j ;
                }
                if(M[i][j]=='H')
                {
                    hh[ch].x=i ;
                    hh[ch++].y=j ;
                }
            }
        } 
        int SS=0 ;
        int ST=2*ch+1 ;
        for(int i=0 ;i<ch ;i++)
        {
            add(0,i+1,1,0) ;
            for(int j=0 ;j<ch ;j++)
            {
                int temp=ABS(mm[i].x-hh[j].x)+ABS(mm[i].y-hh[j].y) ;
                add(i+1,j+1+ch,1,temp) ;
            }
            add(i+1+ch,ST,1,0) ;
        }
        printf("%d
",MCMF(SS,ST,ST+1)) ;
    }
    return 0 ;
}