AmazingCounters.com

POJ 2195 Going Home【最小费用流 二分图最优匹配】

题目大意:一个n*m的地图,上面有一些人man(m)和数量相等的house(H) 图上的距离为曼哈顿距离 问所有人住进一所房子(当然一个人住一间咯)距离之和最短是多少?

思路:一个人一间房,明显是二分图的模型,边权为人和房子的曼哈顿距离,然后算一下最小距离即可 懒得学KM了 最小费用流的经典建图

 

 

#include <stdio.h>

#include <string.h>

#include <algorithm>

#include <queue>

#include <iostream>

#define maxn 40000

#define inf 0x3f3f3f3f

using namespace std;

struct POINT

{

    int x;

    int y;

}a[maxn],b[maxn];

int head[maxn],root[maxn],point[maxn],next[maxn];

int flow[maxn],cost[maxn],pre[maxn],now=0,dist[maxn];

int n,m,h,h2;

char ch[maxn];

int mabs(int x)

{

    return x>0?x:-x;

}

int distanc(int i,int j)

{

    return mabs(a[i].x-b[j].x)+mabs(a[i].y-b[j].y);

}

void add(int x,int y,int v,int c)

{

    next[++now]=head[x];

    head[x]=now;

    point[now]=y;

    flow[now]=v;

    cost[now]=c;

    root[now]=x;

    next[++now]=head[y];

    head[y]=now;

    point[now]=x;

    flow[now]=0;

    cost[now]=-c;

    root[now]=y;

}

int spfa(int s,int t)

{

    memset(pre,0,sizeof(pre));

    for(int i=0;i<=t;i++)dist[i]=inf;

    dist[s]=0;

    int visit[maxn]={0};

    visit[s]=1;

    queue<int>q;

    q.push(s);

    while(!q.empty())

    {

        int u=q.front();

        q.pop();

        visit[u]=0;

        for(int i=head[u];i;i=next[i])

        {

            int k=point[i];

            if(dist[u]+cost[i]<dist[k] && flow[i]!=0)

            {

                pre[k]=i;

                dist[k]=dist[u]+cost[i];

                if(!visit[k])

                {

                    visit[k]=1;

                    q.push(k);

                }

            }

        }

    }

    return dist[t]!=inf;

}

int main()

{

    while(1)

    {

        scanf("%d%d",&n,&m);

        if(n==0&&m==0)break;

        now=0;

        memset(head,0,sizeof(head));

        int ans=0;

        h=h2=0;

        for(int i=1;i<=n;i++)

        {

            scanf("%s",ch+1);

            for(int j=1;j<=m;j++)

            {

                if(ch[j]=='H')a[++h].x=i,a[h].y=j;

                else if(ch[j]=='m')b[++h2].x=i,b[h2].y=j;

            }

        }

        for(int i=1;i<=h;i++)

        {

            for(int j=1;j<=h;j++)

            {

                int u=distanc(i,j);

                add(j,i+h+1,1,u);

            }

        }

        int s=h*2+10,t=h*2+11;

        for(int i=1;i<=h;i++)add(s,i,1,0);

        for(int i=1;i<=h;i++)add(i+h+1,t,1,0);

        while(spfa(s,t))

        {

            int e=pre[t],minx=flow[e];

            while(e)

            {

                if(flow[e]<minx)minx=flow[e];

                e=pre[root[e]];

            }

            e=pre[t];

            while(e)

            {

                flow[e]-=minx;

                flow[((e-1)^1)+1]+=minx;

                e=[root[e]];

            }

            ans+=dist[t]*minx;

        }

        printf("%d\n",ans);

    }

    return 0;

}

 

posted @ 2014-11-17 21:44  philippica  阅读(179)  评论(0编辑  收藏  举报