POJ2195 Going Home(SPFA最小费用最大流)

最新推荐文章于 2021-11-30 21:24:48 发布

xtttgo

最新推荐文章于 2021-11-30 21:24:48 发布

阅读量407

点赞数

分类专栏：网络流

本文链接：https://blog.csdn.net/xtttgo/article/details/50774452

版权

网络流专栏收录该内容

14 篇文章 0 订阅

订阅专栏

链接：http://poj.org/problem?id=2195
题意：一个n*m的矩阵里，有x个人和x个房子，每个人都要进入一个房子，每个房子只能容纳一个人，每个人移动一格的费用是1，问你x个人全部进入到房子的最小花费是多少。
这题的思路很容易想到，建立一个源点和汇点。汇点到每个人的流量为1，花费为0。每个人到每个房子的距离都处理成边，流量为1。每个房子到汇点的流量为1，花费为0。然后一个裸的费用流就出现了。两个点的距离是abs( a.x - b.x ) + abs( a.y - b.y )。
Accepted 1168K 94MS

#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<map>
#include<iostream>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<stack>
#include<cctype>
#include<set>
#include<ctime>
#include<cassert>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int INF=1e9+7;
#define MP make_pair
#define PB push_back
#define X first
#define Y second

struct EE{
    int from,to,cap,cost;
    EE(){}
    EE(int from,int to,int cap,int cost):from(from),to(to),cap(cap),cost(cost){}
}edge[210*210*4];
int n,m,st,ed,idcnt,Ecnt,d[210],pre[210],flow[210];
char pic[110][110];
bool vis[210];
vector<pii> man,house;
vector<vector<int> >G;
map<pii,int> ids;

inline void add(int from,int to,int cap,int cost){
    G[from].PB(Ecnt);
    edge[Ecnt++]=EE(from,to,cap,cost);
    G[to].PB(Ecnt);
    edge[Ecnt++]=EE(to,from,0,-cost);
}
inline int ID(pii a){
    if(ids.count(a))return ids[a];
    return ids[a]=idcnt++;
}
inline int dis(pii a,pii b){//距离
    return abs(a.X-b.X)+abs(a.Y-b.Y);
}
void build(){
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(pic[i][j]=='.')continue;
            if(pic[i][j]=='m'){
                man.PB(MP(i,j));
                ID(man.back());//编号
            }
            else{
                house.PB(MP(i,j));
                ID(house.back());//编号
            }
        }
    }
    st=0;ed=idcnt;
    G.clear();G.resize(ed+1);
    for(int i=0;i<(int)man.size();i++){
        int s=ID(man[i]);
        add(st,s,1,0);
        add(ID(house[i]),ed,1,0);
        for(int j=0;j<(int)house.size();j++){
            int t=ID(house[j]);
            add(s,t,1,dis(man[i],house[j]));
        }
    }
}
bool spfa(int &ff,int &ans){
    queue<int> Q;Q.push(st);
    memset(vis,false,sizeof vis);
    fill(d,d+ed+1,INF);
    d[st]=0;vis[st]=true;pre[st]=-1;flow[st]=INF;
    while(!Q.empty()){
        int s=Q.front();Q.pop();
        vis[s]=false;
        for(int i=0;i<(int)G[s].size();i++){
            EE e=edge[G[s][i]];
            if(e.cap>0&&d[e.to]>d[s]+e.cost){
                d[e.to]=d[s]+e.cost;
                pre[e.to]=G[s][i];
                flow[e.to]=min(flow[s],e.cap);
                if(!vis[e.to]){vis[e.to]=true;Q.push(e.to);}
            }
        }
    }
    if(d[ed]==INF)return false;
    ff+=flow[ed];ans+=flow[ed]*d[ed];
    int now=ed;
    while(now!=st){
        now=pre[now];
        edge[now].cap-=flow[ed];
        edge[now^1].cap+=flow[ed];
        now=edge[now].from;
    }
    return true;
}
void max_flow(){
    int ff=0,ans=0;
    while(spfa(ff,ans));
    printf("%d\n",ans);
}
int main()
{
//    freopen("D://input.txt","r",stdin);
    while(scanf("%d%d",&n,&m)!=EOF&&n&&m){
        ids.clear();man.clear();house.clear();
        idcnt=1;Ecnt=0;
        for(int i=1;i<=n;i++){
            scanf("%s",pic[i]+1);
        }
        build();
        max_flow();
    }
    return 0;
}